Fast differentiable Mean Shift on the GPU using PyKeOps and pytorch.
Features:
- Gaussian, Flat and Epanechnikov Kernels
- Blurring and non-blurring versions
Only the shifting of points is implemented, not the final clustering.
git clone https://github.com/imagirom/gpushift.git
cd gpushift
python setup.py install
Check out Examples.ipynb