提交 937e0756 编写于 作者: Z zhiboniu 提交者: zhiboniu

filter empty features

上级 8bdcd157
...@@ -321,6 +321,8 @@ def res2dict(multi_res): ...@@ -321,6 +321,8 @@ def res2dict(multi_res):
for tid, res in c_res.items(): for tid, res in c_res.items():
key = "c" + str(cid) + "_t" + str(tid) key = "c" + str(cid) + "_t" + str(tid)
if key not in cid_tid_dict: if key not in cid_tid_dict:
if len(res["rects"]) < 10:
continue
cid_tid_dict[key] = res cid_tid_dict[key] = res
cid_tid_dict[key]['mean_feat'] = distill_idfeat(res) cid_tid_dict[key]['mean_feat'] = distill_idfeat(res)
return cid_tid_dict return cid_tid_dict
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册