From d97c6860181c3423a2c007c18a97c67a48e8d7e7 Mon Sep 17 00:00:00 2001 From: Martin <1500595+bmmtstb@users.noreply.github.com> Date: Wed, 18 Oct 2023 12:19:22 +0200 Subject: [PATCH] allow debug flag to change tracker behavior --- trackers/__init__.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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