requirements.txt 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. # Ultralytics requirements
  2. # Example: pip install -r requirements.txt
  3. # Base ----------------------------------------
  4. matplotlib>=3.2.2
  5. numpy>=1.22.2 # pinned by Snyk to avoid a vulnerability
  6. opencv-python>=4.6.0
  7. pillow>=7.1.2
  8. pyyaml>=5.3.1
  9. requests>=2.23.0
  10. scipy>=1.4.1
  11. # torch>=1.8.0
  12. # torchvision>=0.9.0
  13. tqdm>=4.64.0
  14. # Logging -------------------------------------
  15. # tensorboard>=2.13.0
  16. # dvclive>=2.12.0
  17. # clearml
  18. # comet
  19. # Plotting ------------------------------------
  20. pandas>=1.1.4
  21. seaborn>=0.11.0
  22. # Export --------------------------------------
  23. # coremltools>=7.0.b1 # CoreML export
  24. # onnx>=1.12.0 # ONNX export
  25. # onnxsim>=0.4.1 # ONNX simplifier
  26. # nvidia-pyindex # TensorRT export
  27. # nvidia-tensorrt # TensorRT export
  28. # scikit-learn==0.19.2 # CoreML quantization
  29. # tensorflow>=2.4.1 # TF exports (-cpu, -aarch64, -macos)
  30. # tflite-support
  31. # tensorflowjs>=3.9.0 # TF.js export
  32. # openvino-dev>=2023.0 # OpenVINO export
  33. # Extras --------------------------------------
  34. psutil # system utilization
  35. py-cpuinfo # display CPU info
  36. # thop>=0.1.1 # FLOPs computation
  37. # ipython # interactive notebook
  38. # albumentations>=1.0.3 # training augmentations
  39. # pycocotools>=2.0.6 # COCO mAP
  40. # roboflow