YONG627 commited on
Commit
0f61a85
·
1 Parent(s): 60523a3

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +20 -27
requirements.txt CHANGED
@@ -1,49 +1,42 @@
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.18.5
8
- opencv-python>=4.1.1
9
  Pillow>=7.1.2
10
- psutil # system resources
11
  PyYAML>=5.3.1
12
  requests>=2.23.0
13
  scipy>=1.4.1
14
- thop>=0.1.1 # FLOPs computation
15
- torch>=1.7.0 # see https://pytorch.org/get-started/locally (recommended)
16
  torchvision>=0.8.1
17
  tqdm>=4.64.0
18
- ultralytics>=8.0.111
19
- # protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012
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 # CoreML export
32
- # onnx>=1.10.0 # ONNX export
33
- # onnx-simplifier>=0.4.1 # ONNX simplifier
34
  # nvidia-pyindex # TensorRT export
35
  # nvidia-tensorrt # TensorRT export
36
- # scikit-learn<=1.1.2 # CoreML quantization
37
- # tensorflow>=2.4.0 # TF exports (-cpu, -aarch64, -macos)
 
38
  # tensorflowjs>=3.9.0 # TF.js export
39
- # openvino-dev # OpenVINO export
40
 
41
- # Deploy ----------------------------------------------------------------------
42
- setuptools>=65.5.1 # Snyk vulnerability fix
43
- # tritonclient[all]~=2.24.0
44
-
45
- # Extras ----------------------------------------------------------------------
46
  # ipython # interactive notebook
47
- # mss # screenshots
48
  # albumentations>=1.0.3
49
  # pycocotools>=2.0.6 # COCO mAP
 
 
1
+ # Ultralytics requirements
2
  # Usage: pip install -r requirements.txt
3
 
4
+ # Base ----------------------------------------
5
+ matplotlib>=3.2.2
6
+ opencv-python>=4.6.0
 
 
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.64.0
 
 
14
 
15
+ # Logging -------------------------------------
16
+ # tensorboard>=2.13.0
17
+ # clearml
18
  # comet
19
 
20
+ # Plotting ------------------------------------
21
  pandas>=1.1.4
22
  seaborn>=0.11.0
23
 
24
+ # Export --------------------------------------
25
  # coremltools>=6.0 # CoreML export
26
+ # onnx>=1.12.0 # ONNX export
27
+ # onnxsim>=0.4.1 # ONNX simplifier
28
  # nvidia-pyindex # TensorRT export
29
  # nvidia-tensorrt # TensorRT export
30
+ # scikit-learn==0.19.2 # CoreML quantization
31
+ # tensorflow>=2.4.1 # TF exports (-cpu, -aarch64, -macos)
32
+ # tflite-support
33
  # tensorflowjs>=3.9.0 # TF.js export
34
+ # openvino-dev>=2022.3 # OpenVINO export
35
 
36
+ # Extras --------------------------------------
37
+ psutil # system utilization
38
+ # thop>=0.1.1 # FLOPs computation
 
 
39
  # ipython # interactive notebook
 
40
  # albumentations>=1.0.3
41
  # pycocotools>=2.0.6 # COCO mAP
42
+ # roboflow