Update requirements.txt

requirements.txt

@@ -1,42 +1,52 @@
-# pip install -r requirements.txt
+# Ultralytics requirements
+# Example: pip install -r requirements.txt
 
-# base ----------------------------------------
-matplotlib>=3.2.2
-numpy>=1.18.5
-opencv-python-headless
-Pillow
-PyYAML>=5.3.1
+# Base ----------------------------------------
+matplotlib>=3.3.0
+numpy>=1.22.2 # pinned by Snyk to avoid a vulnerability
+opencv-python>=4.6.0
+pillow>=7.1.2
+pyyaml>=5.3.1
+requests>=2.23.0
 scipy>=1.4.1
-torch>=1.8.1
-torchvision>=0.8.1
-tqdm>=4.41.0
-ultralytics>=8.0.116
-psutil==5.9.5
-opencv-python==4.7.0.72
-pandas==2.0.2
+torch>=1.8.0
+torchvision>=0.9.0
+tqdm>=4.64.0
 
-# logging -------------------------------------
-tensorboard>=2.4.1
-# wandb
+# Logging -------------------------------------
+# tensorboard>=2.13.0
+# dvclive>=2.12.0
+# clearml
+# comet
 
-# plotting ------------------------------------
+# Plotting ------------------------------------
+pandas>=1.1.4
 seaborn>=0.11.0
-pandas
 
-# export --------------------------------------
-# coremltools>=4.1
-# onnx>=1.9.0
-# scikit-learn==0.19.2  # for coreml quantization
+# Export --------------------------------------
+# coremltools>=7.0  # CoreML export
+# onnx>=1.12.0  # ONNX export
+# onnxsim>=0.4.1  # ONNX simplifier
+# nvidia-pyindex  # TensorRT export
+# nvidia-tensorrt  # TensorRT export
+# scikit-learn==0.19.2  # CoreML quantization
+# tensorflow>=2.4.1  # TF exports (-cpu, -aarch64, -macos)
+# tflite-support
+# tensorflowjs>=3.9.0  # TF.js export
+# openvino-dev>=2023.0  # OpenVINO export
 
-# extras --------------------------------------
-# Cython  # for pycocotools https://github.com/cocodataset/cocoapi/issues/172
-# pycocotools>=2.0  # COCO mAP
-# albumentations>=1.0.3
-thop  # FLOPs computation
+# Extras --------------------------------------
+psutil  # system utilization
+py-cpuinfo  # display CPU info
+thop>=0.1.1  # FLOPs computation
+# ipython  # interactive notebook
+# albumentations>=1.0.3  # training augmentations
+# pycocotools>=2.0.6  # COCO mAP
+# roboflow
 
 gdown
 
-yolov5==7.0.2
+yolov5
 gradio
 torch
-gradio_client==0.2.7
+gradio_client