web3slinger commited on
Commit
8f7a087
·
1 Parent(s): 20873ac

fix: add dependencies

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