Meanshift Clustering in Pytorch

Thanks David, I really like the idea of using (MiniBatch)KMeans first and then applying MeanShift to the centroids. This reduces the dimensionality of the problem to something that can be handled easily and quickly with a vectorised version of Meanshift. I have been been able to segment high resolution images on just a CPU.

@quortil would love to see your code - sounds great!

Hi Surya,

I wanted to cluster images using meanshift clustering algorithm.Can we apply lsh for an image(single image) dataset?