rachana219 commited on
Commit
8d0a38e
·
1 Parent(s): 6394599

added log statement to log the inputs provided for tracking

Browse files
Files changed (1) hide show
  1. track.py +3 -0
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