yolov6 / requirements.txt
julien-c's picture
julien-c HF staff
Update requirements.txt (#4)
751dc75
huggingface_hub
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