Spaces:
Runtime error
Runtime error
Update requirements.txt
Browse files- requirements.txt +17 -11
requirements.txt
CHANGED
@@ -1,6 +1,3 @@
|
|
1 |
-
gradio
|
2 |
-
huggingface_hub
|
3 |
-
datasets
|
4 |
# YOLOv5 requirements
|
5 |
# Usage: pip install -r requirements.txt
|
6 |
|
@@ -12,35 +9,44 @@ 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 |
-
#
|
|
|
23 |
|
24 |
# Plotting ------------------------------------
|
25 |
pandas>=1.1.4
|
26 |
seaborn>=0.11.0
|
27 |
|
28 |
# Export --------------------------------------
|
29 |
-
# coremltools>=
|
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
|
35 |
-
# tensorflow>=2.4.1 #
|
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 |
-
|
|
|
|
|
|
|
|
|
1 |
# YOLOv5 requirements
|
2 |
# Usage: pip install -r requirements.txt
|
3 |
|
|
|
9 |
PyYAML>=5.3.1
|
10 |
requests>=2.23.0
|
11 |
scipy>=1.4.1
|
12 |
+
torch>=1.7.0 # see https://pytorch.org/get-started/locally/ (recommended)
|
13 |
torchvision>=0.8.1
|
14 |
tqdm>=4.64.0
|
15 |
+
# protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012
|
16 |
|
17 |
# Logging -------------------------------------
|
18 |
tensorboard>=2.4.1
|
19 |
+
# clearml>=1.2.0
|
20 |
+
# comet
|
21 |
|
22 |
# Plotting ------------------------------------
|
23 |
pandas>=1.1.4
|
24 |
seaborn>=0.11.0
|
25 |
|
26 |
# Export --------------------------------------
|
27 |
+
# coremltools>=6.0 # CoreML export
|
28 |
# onnx>=1.9.0 # ONNX export
|
29 |
# onnx-simplifier>=0.4.1 # ONNX simplifier
|
30 |
# nvidia-pyindex # TensorRT export
|
31 |
# nvidia-tensorrt # TensorRT export
|
32 |
+
# scikit-learn<=1.1.2 # CoreML quantization
|
33 |
+
# tensorflow>=2.4.1 # TF exports (-cpu, -aarch64, -macos)
|
34 |
# tensorflowjs>=3.9.0 # TF.js export
|
35 |
# openvino-dev # OpenVINO export
|
36 |
|
37 |
+
# Deploy --------------------------------------
|
38 |
+
# tritonclient[all]~=2.24.0
|
39 |
+
|
40 |
# Extras --------------------------------------
|
41 |
ipython # interactive notebook
|
42 |
psutil # system utilization
|
43 |
thop>=0.1.1 # FLOPs computation
|
44 |
+
gradio
|
45 |
+
huggingface_hub
|
46 |
+
datasets
|
47 |
+
joblib
|
48 |
+
Pillow
|
49 |
+
# mss # screenshots
|
50 |
# albumentations>=1.0.3
|
51 |
# pycocotools>=2.0 # COCO mAP
|
52 |
+
# roboflow
|
|