From 4e2546e344c4cc17dcbe5b2f6a0f63f428435da5 Mon Sep 17 00:00:00 2001 From: XYZ <1290573099@qq.com> Date: Tue, 25 Oct 2022 15:54:42 +0800 Subject: [PATCH] fix bug for mtmct vis (#7181) --- deploy/pipeline/pphuman/mtmct.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/deploy/pipeline/pphuman/mtmct.py b/deploy/pipeline/pphuman/mtmct.py index 9c434903c..bb45432a0 100644 --- a/deploy/pipeline/pphuman/mtmct.py +++ b/deploy/pipeline/pphuman/mtmct.py @@ -155,12 +155,16 @@ def save_mtmct_vis_results(camera_results, captures, output_dir, all_attr_result ): # at least one cid_tid[attrs] is not None will goes to attrs_vis attr_res = [] + cid_str = 'c' + str(cid - 1) + "_" for k in tid_list: + if not k.startswith(cid_str): + continue if (frame_id - 1) >= len(multi_res[k]['attrs']): t_attr = None else: t_attr = multi_res[k]['attrs'][frame_id - 1] attr_res.append(t_attr) + assert len(attr_res) == len(boxes) image = visualize_attr( image, attr_res, boxes, is_mtmct=True) @@ -347,7 +351,7 @@ def res2dict(multi_res): for tid, res in c_res.items(): key = "c" + str(cid) + "_t" + str(tid) if key not in cid_tid_dict: - if len(res["features"])==0: + if len(res["features"]) == 0: continue cid_tid_dict[key] = res cid_tid_dict[key]['mean_feat'] = distill_idfeat(res) -- GitLab