Spaces:
Running
Running
# pip install -r requirements.txt | |
# base ---------------------------------------- | |
ultralytics | |
#gitpython>=3.1.30 | |
gradio | |
IPython | |
matplotlib>=3.2.2 | |
numpy>=1.18.5 | |
opencv-python-headless | |
pillow | |
psutil | |
PyYAML>=5.3.1 | |
scipy>=1.4.1 | |
torch>=1.7.0 | |
torchvision>=0.8.1 | |
tqdm>=4.41.0 | |
# logging ------------------------------------- | |
tensorboard>=2.4.1 | |
# wandb | |
# plotting ------------------------------------ | |
seaborn>=0.11.0 | |
pandas | |
# export -------------------------------------- | |
# coremltools>=4.1 | |
# onnx>=1.9.0 | |
# scikit-learn==0.19.2 # for coreml quantization | |
# extras -------------------------------------- | |
# Cython # for pycocotools https://github.com/cocodataset/cocoapi/issues/172 | |
# pycocotools>=2.0 # COCO mAP | |
# albumentations>=1.0.3 | |
thop # FLOPs computation |