|
a |
|
b/requirements |
|
|
1 |
# YOLOv5 requirements |
|
|
2 |
# Usage: pip install -r requirements.txt |
|
|
3 |
|
|
|
4 |
# Base ------------------------------------------------------------------------ |
|
|
5 |
gitpython>=3.1.30 |
|
|
6 |
matplotlib>=3.3 |
|
|
7 |
numpy>=1.23.5 |
|
|
8 |
opencv-python>=4.1.1 |
|
|
9 |
pillow>=10.3.0 |
|
|
10 |
psutil |
|
|
11 |
PyYAML>=5.3.1 |
|
|
12 |
requests>=2.32.0 |
|
|
13 |
scipy>=1.4.1 |
|
|
14 |
thop>=0.1.1 |
|
|
15 |
torch>=1.8.0 |
|
|
16 |
torchvision>=0.9.0 |
|
|
17 |
tqdm>=4.64.0 |
|
|
18 |
ultralytics>=8.0.232 |
|
|
19 |
# protobuf<=3.20.1 |
|
|
20 |
|
|
|
21 |
# Logging --------------------------------------------------------------------- |
|
|
22 |
# tensorboard>=2.4.1 |
|
|
23 |
# clearml>=1.2.0 |
|
|
24 |
# comet |
|
|
25 |
|
|
|
26 |
# Plotting -------------------------------------------------------------------- |
|
|
27 |
pandas>=1.1.4 |
|
|
28 |
seaborn>=0.11.0 |
|
|
29 |
|
|
|
30 |
# Export ---------------------------------------------------------------------- |
|
|
31 |
# coremltools>=6.0 |
|
|
32 |
# onnx>=1.10.0 |
|
|
33 |
# onnx-simplifier>=0.4.1 |
|
|
34 |
# nvidia-pyindex |
|
|
35 |
# nvidia-tensorrt |
|
|
36 |
# scikit-learn<=1.1.2 |
|
|
37 |
# tensorflow>=2.4.0,<=2.13.1 |
|
|
38 |
# tensorflowjs>=3.9.0 |
|
|
39 |
# openvino-dev>=2023.0 |
|
|
40 |
|
|
|
41 |
# Deploy ---------------------------------------------------------------------- |
|
|
42 |
setuptools>=65.5.1 |
|
|
43 |
|
|
|
44 |
# Extras ---------------------------------------------------------------------- |
|
|
45 |
# ipython |
|
|
46 |
# mss |
|
|
47 |
# albumentations>=1.0.3 |
|
|
48 |
# pycocotools>=2.0.6 |
|
|
49 |
wheel>=0.38.0 |
|
|
50 |
|