|
a |
|
b/yolov5/requirements.txt |
|
|
1 |
# pip install -r requirements.txt |
|
|
2 |
|
|
|
3 |
# Base ---------------------------------------- |
|
|
4 |
matplotlib>=3.2.2 |
|
|
5 |
numpy>=1.18.5 |
|
|
6 |
opencv-python>=4.1.2 |
|
|
7 |
Pillow>=7.1.2 |
|
|
8 |
PyYAML>=5.3.1 |
|
|
9 |
requests>=2.23.0 |
|
|
10 |
scipy>=1.4.1 |
|
|
11 |
torch>=1.7.0 |
|
|
12 |
torchvision>=0.8.1 |
|
|
13 |
tqdm>=4.41.0 |
|
|
14 |
|
|
|
15 |
# Logging ------------------------------------- |
|
|
16 |
tensorboard>=2.4.1 |
|
|
17 |
# wandb |
|
|
18 |
|
|
|
19 |
# Plotting ------------------------------------ |
|
|
20 |
pandas>=1.1.4 |
|
|
21 |
seaborn>=0.11.0 |
|
|
22 |
|
|
|
23 |
# Export -------------------------------------- |
|
|
24 |
# coremltools>=4.1 # CoreML export |
|
|
25 |
# onnx>=1.9.0 # ONNX export |
|
|
26 |
# onnx-simplifier>=0.3.6 # ONNX simplifier |
|
|
27 |
# scikit-learn==0.19.2 # CoreML quantization |
|
|
28 |
# tensorflow>=2.4.1 # TFLite export |
|
|
29 |
# tensorflowjs>=3.9.0 # TF.js export |
|
|
30 |
|
|
|
31 |
# Extras -------------------------------------- |
|
|
32 |
# albumentations>=1.0.3 |
|
|
33 |
# Cython # for pycocotools https://github.com/cocodataset/cocoapi/issues/172 |
|
|
34 |
# pycocotools>=2.0 # COCO mAP |
|
|
35 |
# roboflow |
|
|
36 |
thop # FLOPs computation |