gpushift

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.

Installation

git clone https://github.com/imagirom/gpushift.git
cd gpushift
python setup.py install

Examples

Check out Examples.ipynb