|
|
|
# pip install -r requirements.txt |
|
# Base ---------------------------------------- |
|
seaborn |
|
ipython |
|
psutil |
|
matplotlib>=3.2.2 |
|
numpy>=1.18.5 |
|
opencv-python==4.7.0.68 |
|
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 |
|
joblib |
|
# Logging ------------------------------------- |
|
#tensorboard>=2.4.1 |
|
#wandb |
|
|
|
# Plotting ------------------------------------ |
|
pandas>=1.1.4 |
|
#seaborn>=0.11.0 |
|
|
|
# Export -------------------------------------- |
|
# coremltools>=4.1 # 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 |
|
# tensorflowjs>=3.9.0 # TF.js export |
|
# openvino-dev # OpenVINO export |
|
|
|
# Extras -------------------------------------- |
|
# albumentations>=1.0.3 |
|
# Cython # for pycocotools https://github.com/cocodataset/cocoapi/issues/172 |
|
# pycocotools>=2.0 # COCO mAP |
|
# roboflow |
|
#thop # FLOPs computation |
|
|
|
|
|
# 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 # system resources |
|
PyYAML>=5.3.1 |
|
requests>=2.23.0 |
|
scipy>=1.4.1 |
|
thop>=0.1.1 # FLOPs computation |
|
torch>=1.8.0 # see https://pytorch.org/get-started/locally (recommended) |
|
torchvision>=0.9.0 |
|
tqdm>=4.64.0 |
|
ultralytics>=8.0.232 |
|
# protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012 |
|
|
|
# Logging --------------------------------------------------------------------- |
|
# tensorboard>=2.4.1 |
|
# clearml>=1.2.0 |
|
# comet |
|
|
|
# Plotting -------------------------------------------------------------------- |
|
pandas>=1.1.4 |
|
seaborn>=0.11.0 |
|
|
|
# Export ---------------------------------------------------------------------- |
|
# coremltools>=6.0 # CoreML export |
|
# onnx>=1.10.0 # ONNX 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.0,<=2.13.1 # TF exports (-cpu, -aarch64, -macos) |
|
# tensorflowjs>=3.9.0 # TF.js export |
|
# openvino-dev>=2023.0 # OpenVINO export |
|
|
|
# Deploy ---------------------------------------------------------------------- |
|
setuptools>=65.5.1 # Snyk vulnerability fix |
|
# tritonclient[all]~=2.24.0 |
|
|
|
# Extras ---------------------------------------------------------------------- |
|
# ipython # interactive notebook |
|
# mss # screenshots |
|
# albumentations>=1.0.3 |
|
# pycocotools>=2.0.6 # COCO mAP |
|
wheel>=0.38.0 # not directly required, pinned by Snyk to avoid a vulnerability |
|
|