Rodrigo Prado commited on
Commit
8382f83
1 Parent(s): 7921b4a

adding requirements file.

Browse files
Files changed (1) hide show
  1. requirements.txt +46 -0
requirements.txt ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ gradio
2
+ huggingface_hub
3
+ datasets
4
+ # YOLOv5 requirements
5
+ # Usage: pip install -r requirements.txt
6
+
7
+ # Base ----------------------------------------
8
+ matplotlib>=3.2.2
9
+ numpy>=1.18.5
10
+ opencv-python>=4.1.1
11
+ Pillow>=7.1.2
12
+ PyYAML>=5.3.1
13
+ requests>=2.23.0
14
+ scipy>=1.4.1
15
+ torch>=1.7.0
16
+ torchvision>=0.8.1
17
+ tqdm>=4.64.0
18
+ protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012
19
+
20
+ # Logging -------------------------------------
21
+ tensorboard>=2.4.1
22
+ # wandb
23
+
24
+ # Plotting ------------------------------------
25
+ pandas>=1.1.4
26
+ seaborn>=0.11.0
27
+
28
+ # Export --------------------------------------
29
+ # coremltools>=5.2 # CoreML export
30
+ # onnx>=1.9.0 # ONNX export
31
+ # onnx-simplifier>=0.4.1 # ONNX simplifier
32
+ # nvidia-pyindex # TensorRT export
33
+ # nvidia-tensorrt # TensorRT export
34
+ # scikit-learn==0.19.2 # CoreML quantization
35
+ # tensorflow>=2.4.1 # TFLite export (or tensorflow-cpu, tensorflow-aarch64)
36
+ # tensorflowjs>=3.9.0 # TF.js export
37
+ # openvino-dev # OpenVINO export
38
+
39
+ # Extras --------------------------------------
40
+ ipython # interactive notebook
41
+ psutil # system utilization
42
+ thop>=0.1.1 # FLOPs computation
43
+ # albumentations>=1.0.3
44
+ # pycocotools>=2.0 # COCO mAP
45
+ # roboflow
46
+