uv install [프로젝트 명]
uv run main.py
uv add open-clip-torch torchvision Pillow
open-clip-torch
OpenCLIP 모델을 PyTorch 기반으로 쉽게 사용할 수 있게 해주는 라이브러리
- OpenAI의 CLIP 구조를 재현한 다양한 멀티모달 모델 로딩
- 텍스트와 이미지를 각각 임베딩하고, 서로의 유사도 계산 가능
- Hugging Face처럼 손쉽게 모델 이름, 사전학습 버전 지정 가능
torchvision
PyTorch의 컴퓨터 비전 기능을 지원하는 라이브러리
- 이미지 전처리(transform) 함수 제공 (
Resize
, Normalize
, ToTensor
등)
- 이미지 분류/탐지용 샘플 데이터셋 내장 (e.g. CIFAR-10, ImageNet 등)
- 모델(ResNet, MobileNet 등)도 함께 제공
Pillow
Python에서 이미지 파일을 열고, 저장하고, 변형하는 라이브러리
- PNG, JPEG, BMP 등 다양한 포맷 지원
- 이미지 크기 조정, 자르기, 회전 등 기본 조작 가능
torchvision
과 자주 함께 사용됨
https://huggingface.co/docs/hub/open_clip