SegMatch / requirements.txt
skallewag's picture
Update requirements.txt
7fa3259 verified
raw
history blame contribute delete
350 Bytes
torch==2.1.0
torchaudio
torchvision
tf-keras
transformers==4.42.4
diffusers
opencv-python
Pillow
numpy
matplotlib
scipy
scikit-learn
# For Gradio interface
gradio
# Face comparison
deepface
# Human parts segmentation
onnxruntime
# Clothing segmentation
huggingface-hub>=0.19.0
segment-anything>=1.0
# Color matching dependencies
color-matcher