SWClassifier / requirements.txt
tcooper-xx's picture
Initial Commit
34ecf0d
# Web server
fastapi>=0.110.0
uvicorn[standard]>=0.29.0
python-multipart>=0.0.9
# ML / vision (already present in classification_model/requirements.txt)
torch>=1.10.2
torchvision>=0.11.3
numpy>=1.19.2
opencv-python
Pillow>=8.4.0
ultralytics
faiss-cpu
scipy
scikit-learn
timm==1.0.15
shapely
albumentations==2.0.8