Meanshift Clustering in Pytorch

(Gavin Francis) #41

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.

(Jeremy Howard) #42

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

(sindhuja vakkalagadda) #43

Hi Surya,

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