brian25 commited on
Commit
8b0c9a0
·
verified ·
1 Parent(s): d2b9f29

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -49
requirements.txt CHANGED
@@ -1,49 +1,7 @@
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 # system resources
11
- PyYAML>=5.3.1
12
- requests>=2.32.2
13
- scipy>=1.4.1
14
- thop>=0.1.1 # FLOPs computation
15
- torch>=1.8.0 # see https://pytorch.org/get-started/locally (recommended)
16
- torchvision>=0.9.0
17
- tqdm>=4.66.3
18
- ultralytics>=8.2.34 # https://ultralytics.com
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,<=2.13.1 # TF exports (-cpu, -aarch64, -macos)
38
- # tensorflowjs>=3.9.0 # TF.js export
39
- # openvino-dev>=2023.0 # OpenVINO export
40
-
41
- # Deploy ----------------------------------------------------------------------
42
- setuptools>=70.0.0 # 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
+ torch>=1.7
2
+ torchvision>=0.8
3
+ gradio>=3.0
4
+ opencv-python
5
+ pandas
6
+ numpy
7
+ PyYAML