Skip to content

Modified CLIP (Contrastive Language-Image Pretraining), Predict the most relevant image given a text.

License

Notifications You must be signed in to change notification settings

WensongVincent/CLIP_modified

 
 

Repository files navigation

Text-Image Retrieval using CLIP

[Paper]

To use the text-image retrieval, run file: Zero-shot text-image retrieval using CLIP.ipynb

Environment setup

$ conda install pytorch pytorch=1.7.1 torchvision cudatoolkit=11.1
$ pip install ftfy regex tqdm
$ pip install git+https://github.com/openai/CLIP.git
$ conda install matplotlib
$ pip install numpy
$ pip install scikit-image

Example result

User given text, retrive image from CIFAR100 dataset

image

User given text, retrie image from sci-kit images

image

About

Modified CLIP (Contrastive Language-Image Pretraining), Predict the most relevant image given a text.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 99.3%
  • Python 0.7%