检索相似图片的算法
通过向量库,检索相似图片,逻辑如下:
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