From 3f855edca51d0fab47badb781fc1029b7448d48d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 24 Sep 2022 13:15:34 +0200 Subject: [PATCH] Update requirements.txt (#1973) --- requirements.txt | 31 +++++++++++++++++++++---------- 1 file changed, 21 insertions(+), 10 deletions(-) diff --git a/requirements.txt b/requirements.txt index e9843e17..7c01cc06 100755 --- a/requirements.txt +++ b/requirements.txt @@ -1,36 +1,47 @@ -# pip install -r requirements.txt +# YOLOv3 requirements +# Usage: pip install -r requirements.txt # Base ---------------------------------------- matplotlib>=3.2.2 numpy>=1.18.5 -opencv-python>=4.1.2 +opencv-python>=4.1.1 Pillow>=7.1.2 PyYAML>=5.3.1 requests>=2.23.0 scipy>=1.4.1 torch>=1.7.0 torchvision>=0.8.1 -tqdm>=4.41.0 +tqdm>=4.64.0 +# protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012 # Logging ------------------------------------- tensorboard>=2.4.1 -wandb +# wandb +# clearml # Plotting ------------------------------------ pandas>=1.1.4 seaborn>=0.11.0 # Export -------------------------------------- -# coremltools>=4.1 # CoreML export +# coremltools>=6.0 # CoreML export # onnx>=1.9.0 # ONNX export -# onnx-simplifier>=0.3.6 # ONNX simplifier -# scikit-learn==0.19.2 # CoreML quantization -# tensorflow>=2.4.1 # TFLite export +# onnx-simplifier>=0.4.1 # ONNX simplifier +# nvidia-pyindex # TensorRT export +# nvidia-tensorrt # TensorRT export +# scikit-learn<=1.1.2 # CoreML quantization +# tensorflow>=2.4.1 # TF exports (-cpu, -aarch64, -macos) # tensorflowjs>=3.9.0 # TF.js export +# openvino-dev # OpenVINO export + +# Deploy -------------------------------------- +# tritonclient[all]~=2.24.0 # Extras -------------------------------------- +ipython # interactive notebook +psutil # system utilization +thop>=0.1.1 # FLOPs computation +# mss # screenshots # albumentations>=1.0.3 -# Cython # for pycocotools https://github.com/cocodataset/cocoapi/issues/172 # pycocotools>=2.0 # COCO mAP # roboflow -thop # FLOPs computation