enhanced_transfer_learning / requirements.txt
AndrzejDD's picture
Update requirements.txt
b4f277e verified
torch==2.0.1
torchvision==0.15.2
ultralytics==8.2
numpy==1.26.4
opencv-python-headless==4.9.0.80
onnxruntime
requests
PyYAML
gradio
onnxslim
onnx
wandb
onnx
onnxruntime-gpu
huggingface-hub
pycocotools
safetensors
psutil
wget
ttach
pyyaml