requirements.txt

# 参考：https://github.com/ultralytics/yolov5/blob/master/requirements.txt
# pip install -r requirements.txt
# pip 22.1.2

# 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


# Logging -------------------------------------
tensorboard
# wandb

# Plotting ------------------------------------
pandas
seaborn

# Export --------------------------------------
coremltools  # CoreML export
onnx  # ONNX export
onnx-simplifier  # ONNX simplifier
onnxruntime
scikit-learn==0.19.2 # CoreML quantization
tensorflow-cpu  # TFLite export
tensorflowjs  # TF.js export
openvino-dev  # OpenVINO export

# Extras --------------------------------------
ipython  # interactive notebook
psutil  # system utilization
thop  # FLOPs computation
# albumentations>=1.0.3
# pycocotools>=2.0  # COCO mAP
# roboflow