erdogant / clustimage

clustimage is a python package for unsupervised clustering of images.
https://erdogant.github.io/clustimage
Other
92 stars 8 forks source link

Extract features using deep learning #19

Open JoeyChallita opened 1 year ago

JoeyChallita commented 1 year ago

Hello! Great library. I'm currently using it to cluster unlabeled images. Clustering using pca or hog did not yield good results so I extracted features for my images from a pre-trained CNN model that I had , then continued with hierarchical clustering using clustimage. However, I had to work around the code in the library to make it work, mainly because I had to initialize the clustimage with pca or hog and then manually replace the results['feat'] with the deep learning features. If you're interested to expanding the library to deep learning features, let me know, I'll update the code and do a pull request!

Regards, Joey

erdogant commented 1 year ago

Great work Joey, I like it! If you want to share it, that would be awesome!

erdogant commented 1 year ago

Hi @JoeyChallita any luck in using CNNs? There is another issue #21 that has a similar question.