Spaces:
Runtime error
Runtime error
Commit
·
8d0a38e
1
Parent(s):
6394599
added log statement to log the inputs provided for tracking
Browse files
track.py
CHANGED
@@ -85,6 +85,8 @@ def run(
|
|
85 |
vid_stride=1, # video frame-rate stride
|
86 |
retina_masks=False,
|
87 |
):
|
|
|
|
|
88 |
|
89 |
source = str(source)
|
90 |
save_img = not nosave and not source.endswith('.txt') # save inference images
|
@@ -108,6 +110,7 @@ def run(
|
|
108 |
# Load model
|
109 |
device = select_device(device)
|
110 |
is_seg = '-seg' in str(yolo_weights)
|
|
|
111 |
model = AutoBackend(yolo_weights, device=device, dnn=dnn, fp16=half)
|
112 |
stride, names, pt = model.stride, model.names, model.pt
|
113 |
imgsz = check_imgsz(imgsz, stride=stride) # check image size
|
|
|
85 |
vid_stride=1, # video frame-rate stride
|
86 |
retina_masks=False,
|
87 |
):
|
88 |
+
#print the inputs
|
89 |
+
print(f"model used : {yolo_weights}, tracking method : {tracking_method}")
|
90 |
|
91 |
source = str(source)
|
92 |
save_img = not nosave and not source.endswith('.txt') # save inference images
|
|
|
110 |
# Load model
|
111 |
device = select_device(device)
|
112 |
is_seg = '-seg' in str(yolo_weights)
|
113 |
+
|
114 |
model = AutoBackend(yolo_weights, device=device, dnn=dnn, fp16=half)
|
115 |
stride, names, pt = model.stride, model.names, model.pt
|
116 |
imgsz = check_imgsz(imgsz, stride=stride) # check image size
|