search

jhgan00 / image-retrieval-transformers

(Unofficial) PyTorch implementation of Training Vision Transformers for Image Retrieval(El-Nouby, Alaaeldin, et al. 2021).
44 stars 6 forks source link

Can you provide test code? #3

Open Jacky-Android opened 1 year ago

Jacky-Android commented 1 year ago

I need to test the code to retrieve similar pictures, if you can provide it, it is very grateful!

jhgan00 commented 1 year ago

The following Colab notebook is an example of searching for similar images using a model trained on CUB200 data. Hope this is helpful.

https://colab.research.google.com/drive/14tv3FjR6RzJ6VsUjJiqguADSIo2Z4S90#scrollTo=_B5iD5hVGpi6