Postprocessing using siamese

OML-Team / open-metric-learning

Metric learning and retrieval pipelines, models and zoo.

https://open-metric-learning.readthedocs.io/en/latest/index.html

Apache License 2.0

890 stars 61 forks source link

Postprocessing using siamese #629

Closed darthvaddor closed 2 weeks ago

darthvaddor commented 3 weeks ago

Hi @AlekseySh! I'm trying to build a image retriever. I have unlabelled images. So I finetuned DINO on them to generate embeddings. I've saved the embeddings in a cluster, and I'm performing cosine similartity to retrieve top 5 most similar images. The results aren't that great so I want to further do reranking. Will STIR: Siamese Transformer for Image Retrieval Postprocessing be a good resource? I'm asking this because Open Metric Learning is completely supervised, and my embedding model isn't

AlekseySh commented 2 weeks ago

Hi, @darthvaddor The whole OML is supervised. As I suggested before, you may find unsupervised methods in: https://github.com/lightly-ai/lightly