Sall-eGarbageDetection / requirements.txt
LiamKhoaLe's picture
Add YOLOv8n to detect animal and 2 Roboflow models to be specifically finetuned on bird and fish. Rm HTML_CONTENT to be rendered on statics dir
57328cd
raw
history blame contribute delete
364 Bytes
# Image Processing
opencv-python-headless
pillow
numpy
pillow-avif-plugin
# Vision Models
torch
torchvision
torchaudio
scikit-learn
ultralytics
timm
yolov5
huggingface_hub>=0.20.3
transformers==4.37.2
accelerate==0.27.2
inference-sdk
# Video Processing
ffmpeg-python
# Server
fastapi
uvicorn
python-multipart
# Git (needed for clone/setup LoveDA repo)
git-lfs