idealo / imagededup

😎 Finding duplicate images made easy!
https://idealo.github.io/imagededup/
Apache License 2.0
5.1k stars 458 forks source link

Adding cnn calculate cosine by gpu #202

Closed pphuc25 closed 1 year ago

pphuc25 commented 1 year ago

Upon examining the code, I noticed that it utilizes the sklearn cosine function to detect duplicates in the CNN. However, this implementation hampers performance, leading to slower execution. I recommend employing PyTorch to compute the cosine similarity instead, which should significantly improve the efficiency.

tanujjain commented 1 year ago

Thanks for the pointer. Would you be interested in making a PR?

pphuc25 commented 1 year ago

Yes!, I am going to do it