Skip to content

This code implements a versatile image search engine leveraging the CLIP model and FAISS, capable of processing both text-to-image and image-to-image queries.

Notifications You must be signed in to change notification settings

NTUYWANG103/clip-image-search

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Introduction

This code implements a versatile image search engine leveraging the CLIP model and FAISS, capable of processing both text-to-image and image-to-image queries.

Image to image

img2img

Text to image

txt2img

Install Environment

pip install -r requirements.txt

Preprocess

python clip_image_search/extract_embeddings.py --img_dir your_dataset_dir --save_path results/embeddings.pkl

python clip_image_search/build_index.py --embeddings_path results/embeddings.pkl --save_path results/index.faiss

Demo

streamlit run app.py

Star History

Star History Chart

About

This code implements a versatile image search engine leveraging the CLIP model and FAISS, capable of processing both text-to-image and image-to-image queries.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages