From 937e07564f9b8f7576c0b08280d3738f44744d5e Mon Sep 17 00:00:00 2001 From: zhiboniu Date: Mon, 25 Apr 2022 07:46:51 +0000 Subject: [PATCH] filter empty features --- deploy/pphuman/mtmct.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/deploy/pphuman/mtmct.py b/deploy/pphuman/mtmct.py index 5c7494edf..7c4d57115 100644 --- a/deploy/pphuman/mtmct.py +++ b/deploy/pphuman/mtmct.py @@ -321,6 +321,8 @@ 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["rects"]) < 10: + continue cid_tid_dict[key] = res cid_tid_dict[key]['mean_feat'] = distill_idfeat(res) return cid_tid_dict -- GitLab