You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

41 lines
1.0KB

  1. # YOLOv5 requirements
  2. # Usage: pip install -r requirements.txt
  3. # Base ----------------------------------------
  4. matplotlib>=3.2.2
  5. numpy>=1.18.5
  6. opencv-python>=4.1.1
  7. Pillow>=7.1.2
  8. PyYAML>=5.3.1
  9. requests>=2.23.0
  10. scipy>=1.4.1 # Google Colab version
  11. torch>=1.7.0
  12. torchvision>=0.8.1
  13. tqdm>=4.41.0
  14. protobuf<4.21.3 # https://github.com/ultralytics/yolov5/issues/8012
  15. # Logging -------------------------------------
  16. tensorboard>=2.4.1
  17. # wandb
  18. # Plotting ------------------------------------
  19. pandas>=1.1.4
  20. seaborn>=0.11.0
  21. # Export --------------------------------------
  22. # coremltools>=4.1 # CoreML export
  23. # onnx>=1.9.0 # ONNX export
  24. # onnx-simplifier>=0.3.6 # ONNX simplifier
  25. # scikit-learn==0.19.2 # CoreML quantization
  26. # tensorflow>=2.4.1 # TFLite export
  27. # tensorflowjs>=3.9.0 # TF.js export
  28. # openvino-dev # OpenVINO export
  29. # Extras --------------------------------------
  30. ipython # interactive notebook
  31. psutil # system utilization
  32. thop # FLOPs computation
  33. # albumentations>=1.0.3
  34. # pycocotools>=2.0 # COCO mAP
  35. # roboflow