Enable filtering by image with clip-retrieval filter

rom1504 / clip-retrieval

Easily compute clip embeddings and build a clip retrieval system with them

https://rom1504.github.io/clip-retrieval/

MIT License

2.42k stars 213 forks source link

Enable filtering by image with clip-retrieval filter #232

Closed resloved closed 1 year ago

resloved commented 1 year ago

Just added a check to see if the query is an image path and using encode_image to generate or embedding to filter by instead of encode_text. Might be worth having a separate flag for querying by image, but this felt the most intuitive to me.