1. Github

GitHub - pharmapsychotic/clip-interrogator: Image to prompt with BLIP and CLIP

2. 환경설정

• conda 가상 환경 만들고 activate

conda create --name clip python=3.8
source activate clip

• 설치

pip3 install torch torchvision --extra-index-url <https://download.pytorch.org/whl/cu117>
pip install clip-interrogator==0.5.4

3. 오류 해결

• cannot import name 'InterpolationMode' from 'torchvision.transforms’ 에러

pip install --upgrade torchvision

• dimension 에러

pip install transformers==4.26.1

4. 코드

img-to-prompt-run.py

img-to-prompt.ipynb

img-to-prompt-run.py