#RePost #AI #Doc #DevOps

检索相似图片的算法

通过向量库,检索相似图片,逻辑如下:

1. 将所有图片向量化,通过OpenAI CLIP库

https://huggingface.co/docs/transformers/model_doc/clip

1. 将向量结果存到向量数据库里面
2. 文字检索

文字=>向量化 =>相似查找HNSW=>结果

代码地址:

https://github.com/pingcap/tidb-vector-python/blob/main/examples/image_search/example.ipynb

Google Colab:

https://colab.research.google.com/github/deep-diver/image_search_with_natural_language/blob/main/notebooks/Image_Search_CLIP.ipynb#scrollTo=q5xcJyqKMu2z

via 人工智能讨论组 - 即刻圈子 (author: 毛毛读书忙)

Message link
 
 
Back to Top
OKHK