Diego Fernandez commited on
Commit
0993fd4
1 Parent(s): 31aaaec

chore: change model to YOLOv7

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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=["YoloV7", "YoloV7 Tiny"], value="YoloV7", label="Model")
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 = {"YoloV7": "custom_models/yolov7.pt", "YoloV7 Tiny": "custom_models/yolov7-tiny.pt"}
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"}