Spaces:
Runtime error
Runtime error
| # YOLOv5 requirements | |
| # Usage: pip install -r requirements.txt | |
| # Base ---------------------------------------- | |
| matplotlib>=3.2.2 | |
| numpy>=1.18.5 | |
| opencv-python>=4.1.1 | |
| Pillow>=7.1.2 | |
| PyYAML>=5.3.1 | |
| requests>=2.23.0 | |
| scipy>=1.4.1 # Google Colab version | |
| torch>=1.7.0,!=1.12.0 # https://github.com/ultralytics/yolov5/issues/8395 | |
| torchvision>=0.8.1,!=0.13.0 # https://github.com/ultralytics/yolov5/issues/8395 | |
| tqdm>=4.41.0 | |
| protobuf<4.21.3 # https://github.com/ultralytics/yolov5/issues/8012 | |
| # Logging ------------------------------------- | |
| tensorboard>=2.4.1 | |
| # wandb | |
| # Plotting ------------------------------------ | |
| pandas>=1.1.4 | |
| seaborn>=0.11.0 | |
| # Export -------------------------------------- | |
| coremltools>=5.2 # CoreML export | |
| onnx>=1.9.0 # ONNX export | |
| onnx-simplifier>=0.4.1 # ONNX simplifier | |
| onnxruntime | |
| # nvidia-pyindex # TensorRT export | |
| # nvidia-tensorrt # TensorRT export | |
| scikit-learn>=0.19.2 # CoreML quantization | |
| tensorflow-gpu>=2.4.1 # TFLite export | |
| tensorflowjs>=3.9.0 # TF.js export | |
| openvino-dev==2022.1.0 # OpenVINO export | |
| # Extras -------------------------------------- | |
| ipython # interactive notebook | |
| psutil # system utilization | |
| thop>=0.1.0 # FLOPs computation | |
| # albumentations>=1.0.3 | |
| # pycocotools>=2.0 # COCO mAP | |
| # roboflow | |