Spaces:
Runtime error
Runtime error
Diego Fernandez
commited on
Commit
•
0993fd4
1
Parent(s):
31aaaec
chore: change model to YOLOv7
Browse files- app.py +1 -1
- demo_utils/configuration.py +1 -1
app.py
CHANGED
@@ -6,7 +6,7 @@ from inference import inference
|
|
6 |
|
7 |
input_video = gr.Video(mirror_webcam=False)
|
8 |
|
9 |
-
dd_model = gr.Dropdown(choices=["
|
10 |
|
11 |
cb_motion_estimation = gr.Checkbox(value=True, label="Track camera movement")
|
12 |
|
|
|
6 |
|
7 |
input_video = gr.Video(mirror_webcam=False)
|
8 |
|
9 |
+
dd_model = gr.Dropdown(choices=["YOLOv7", "YOLOv7 Tiny"], value="YOLOv7", label="Model")
|
10 |
|
11 |
cb_motion_estimation = gr.Checkbox(value=True, label="Track camera movement")
|
12 |
|
demo_utils/configuration.py
CHANGED
@@ -2,6 +2,6 @@ DISTANCE_THRESHOLD_BBOX: float = 3.33
|
|
2 |
DISTANCE_THRESHOLD_CENTROID: int = 30
|
3 |
MAX_DISTANCE: int = 10000
|
4 |
|
5 |
-
models_path = {"
|
6 |
|
7 |
style = {"Bounding box": "bbox", "Centroid": "centroid"}
|
|
|
2 |
DISTANCE_THRESHOLD_CENTROID: int = 30
|
3 |
MAX_DISTANCE: int = 10000
|
4 |
|
5 |
+
models_path = {"YOLOv7": "custom_models/yolov7.pt", "YOLOv7 Tiny": "custom_models/yolov7-tiny.pt"}
|
6 |
|
7 |
style = {"Bounding box": "bbox", "Centroid": "centroid"}
|