app update

requirements.txt

@@ -1,47 +1,43 @@
-# 参考：https://github.com/ultralytics/yolov5/blob/master/requirements.txt
-# pip install -r requirements.txt
-# pip 22.1.2
+# YOLOv5 requirements
+# Usage: pip install -r requirements.txt
 
 # Base ----------------------------------------
-matplotlib
-numpy
-opencv-python
-Pillow
-PyYAML
-requests
-scipy  # Google Colab version
-torch  # https://github.com/ultralytics/yolov5/issues/8395
-torchvision # https://github.com/ultralytics/yolov5/issues/8395
-tqdm
-protobuf  # https://github.com/ultralytics/yolov5/issues/8012
-
-
-# Streamlit YOLOv5 Model2X
-streamlit
-
+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
+tensorboard>=2.4.1
 # wandb
 
 # Plotting ------------------------------------
-pandas
-seaborn
+pandas>=1.1.4
+seaborn>=0.11.0
 
 # Export --------------------------------------
-coremltools  # CoreML export
-onnx  # ONNX export
-onnx-simplifier  # ONNX simplifier
+coremltools>=4.1  # CoreML export
+onnx>=1.9.0  # ONNX export
+onnx-simplifier>=0.3.6  # ONNX simplifier
 onnxruntime
-scikit-learn==0.19.2 # CoreML quantization
-tensorflow-cpu  # TFLite export
-tensorflowjs  # TF.js export
+# 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  # OpenVINO export
 
 # Extras --------------------------------------
 ipython  # interactive notebook
 psutil  # system utilization
-thop  # FLOPs computation
+thop>=0.1.0  # FLOPs computation
 # albumentations>=1.0.3
 # pycocotools>=2.0  # COCO mAP
 # roboflow