diff --git a/deploy/pipeline/pipeline.py b/deploy/pipeline/pipeline.py index 0f675ef149370346db9867628394ed3666751fc1..fe3654dae23620a4f40b25ff7431ed34243e40a7 100644 --- a/deploy/pipeline/pipeline.py +++ b/deploy/pipeline/pipeline.py @@ -966,6 +966,7 @@ class PipePredictor(object): online_scores, frame_id=frame_id, fps=fps, + ids2names=self.mot_predictor.pred_config.labels, do_entrance_counting=self.do_entrance_counting, do_break_in_counting=self.do_break_in_counting, entrance=entrance, diff --git a/deploy/pptracking/python/mot/visualize.py b/deploy/pptracking/python/mot/visualize.py index 141140f8ed8cc30d1b057bbc1093abd5074bfccc..9a4cb5806c769fb8c97177cfd9b06a1163f78944 100644 --- a/deploy/pptracking/python/mot/visualize.py +++ b/deploy/pptracking/python/mot/visualize.py @@ -191,7 +191,7 @@ def plot_tracking_dict(image, scores_dict, frame_id=0, fps=0., - ids2names=['pedestrian'], + ids2names=[], do_entrance_counting=False, do_break_in_counting=False, entrance=None, diff --git a/deploy/pptracking/python/mot_sde_infer.py b/deploy/pptracking/python/mot_sde_infer.py index 1e92652fd7a44541acb3e325cf2cbc6db872cb1c..b62905e2aecfc4aaacb8bfc452198b50bf94ef88 100644 --- a/deploy/pptracking/python/mot_sde_infer.py +++ b/deploy/pptracking/python/mot_sde_infer.py @@ -512,14 +512,15 @@ class SDE_Detector(Detector): online_ids, online_scores, frame_id=frame_id, - ids2names=[]) + ids2names=ids2names) else: im = plot_tracking( frame, online_tlwhs, online_ids, online_scores, - frame_id=frame_id) + frame_id=frame_id, + ids2names=ids2names) save_dir = os.path.join(self.output_dir, seq_name) if not os.path.exists(save_dir): os.makedirs(save_dir) @@ -632,6 +633,7 @@ class SDE_Detector(Detector): online_scores, frame_id=frame_id, fps=fps, + ids2names=ids2names, do_entrance_counting=self.do_entrance_counting, entrance=entrance) else: diff --git a/deploy/python/mot_sde_infer.py b/deploy/python/mot_sde_infer.py index 29b80751f94ee0fd954dcdb2bf1f6a55810c9ab5..b4a487facddc4a6eb9492ba367c5333b0c77d9a9 100644 --- a/deploy/python/mot_sde_infer.py +++ b/deploy/python/mot_sde_infer.py @@ -359,14 +359,15 @@ class SDE_Detector(Detector): online_ids, online_scores, frame_id=frame_id, - ids2names=[]) + ids2names=ids2names) else: im = plot_tracking( frame, online_tlwhs, online_ids, online_scores, - frame_id=frame_id) + frame_id=frame_id, + ids2names=ids2names) save_dir = os.path.join(self.output_dir, seq_name) if not os.path.exists(save_dir): os.makedirs(save_dir) @@ -431,7 +432,8 @@ class SDE_Detector(Detector): online_ids, online_scores, frame_id=frame_id, - fps=fps) + fps=fps, + ids2names=ids2names) else: # use ByteTracker, support multiple class for cls_id in range(num_classes):