model-optimization-lab / requirements.txt
shriarul5273's picture
add YOLO12 models and update README, app.py, and requirements.txt
2797bac
raw
history blame contribute delete
339 Bytes
# Core vision + models
torch>=2.2.0
torchvision>=0.17.0
timm>=0.9.12
segmentation-models-pytorch>=0.3.3
huggingface-hub>=0.23.0
albumentations>=1.4.8
ultralytics>=8.3.0 # YOLO12 detection backends
# UI
gradio>=4.19.2
# Data + plotting
pandas>=2.1.0
matplotlib>=3.7.0
numpy>=1.24.0
pillow>=10.0.0
# Optional export targets
onnx>=1.14.0