--- a +++ b/requirements @@ -0,0 +1,50 @@ +# YOLOv5 requirements +# Usage: pip install -r requirements.txt + +# Base ------------------------------------------------------------------------ +gitpython>=3.1.30 +matplotlib>=3.3 +numpy>=1.23.5 +opencv-python>=4.1.1 +pillow>=10.3.0 +psutil +PyYAML>=5.3.1 +requests>=2.32.0 +scipy>=1.4.1 +thop>=0.1.1 +torch>=1.8.0 +torchvision>=0.9.0 +tqdm>=4.64.0 +ultralytics>=8.0.232 +# protobuf<=3.20.1 + +# Logging --------------------------------------------------------------------- +# tensorboard>=2.4.1 +# clearml>=1.2.0 +# comet + +# Plotting -------------------------------------------------------------------- +pandas>=1.1.4 +seaborn>=0.11.0 + +# Export ---------------------------------------------------------------------- +# coremltools>=6.0 +# onnx>=1.10.0 +# onnx-simplifier>=0.4.1 +# nvidia-pyindex +# nvidia-tensorrt +# scikit-learn<=1.1.2 +# tensorflow>=2.4.0,<=2.13.1 +# tensorflowjs>=3.9.0 +# openvino-dev>=2023.0 + +# Deploy ---------------------------------------------------------------------- +setuptools>=65.5.1 + +# Extras ---------------------------------------------------------------------- +# ipython +# mss +# albumentations>=1.0.3 +# pycocotools>=2.0.6 +wheel>=0.38.0 +