https://github.com/isarandi/tensorflow-mean-shift

Mean-shift clustering algorithm in TensorFlow
https://github.com/isarandi/tensorflow-mean-shift

clustering mean-shift meanshift parallel tensorflow

Last synced: about 1 year ago
JSON representation

Mean-shift clustering algorithm in TensorFlow

• Host: GitHub
• URL: https://github.com/isarandi/tensorflow-mean-shift
• Owner: isarandi
• License: gpl-3.0
• Created: 2019-05-09T16:11:07.000Z (almost 7 years ago)
• Default Branch: master
• Last Pushed: 2021-11-04T18:38:27.000Z (over 4 years ago)
• Last Synced: 2025-01-16T15:35:58.872Z (about 1 year ago)
• Topics: clustering, mean-shift, meanshift, parallel, tensorflow
• Language: Python
• Size: 16.6 KB
• Stars: 0
• Watchers: 2
• Forks: 2
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# Mean shift clustering with TensorFlow

This is a parallelized implementation of the [mean shift](https://en.wikipedia.org/wiki/Mean_shift) clustering algorithm through TensorFlow.

This makes it easy to use on the GPU if you install TensorFlow with CUDA support.

I wrote this to better understand `tf.while_loop` and how to implement a nontrivial iterative algorithm in a TensorFlow graph. It only works in TensorFlow v1, or through the `tensorflow.compat.v1` module of TF2.