/gpushift

Mean Shift on the GPU using PyKeOps and pytorch

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

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