diff --git a/trackers/__init__.py b/trackers/__init__.py index d05eff15..092ec533 100644 --- a/trackers/__init__.py +++ b/trackers/__init__.py @@ -1,8 +1,7 @@ -import numpy as np import torch def track(tracker,args,orig_img,inps,boxes,hm,cropped_boxes,im_name,scores): hm = hm.cpu().data.numpy() - online_targets = tracker.update(orig_img,inps,boxes,hm,cropped_boxes,im_name,scores,_debug=False) + online_targets = tracker.update(orig_img, inps, boxes, hm, cropped_boxes, im_name, scores, _debug=args.debug) new_boxes,new_scores,new_ids,new_hm,new_crop = [],[],[],[],[] for t in online_targets: tlbr = t.tlbr