yolov6 / requirements.txt
nateraw's picture
Update requirements.txt
b96f016
raw
history blame
296 Bytes
huggingface_hub
gradio==3.1.0
torch>=1.8.0
torchvision>=0.9.0
numpy>=1.18.5
opencv-python>=4.1.2
PyYAML>=5.3.1
scipy>=1.4.1
# tqdm>=4.41.0
# addict>=2.4.0
# tensorboard>=2.7.0
# pycocotools>=2.0
# onnx>=1.10.0 # ONNX export
# onnx-simplifier>=0.3.6 # ONNX simplifier
# thop # FLOPs computation