未验证 提交 d5a5fa4b 编写于 作者: W wangguanzhong 提交者: GitHub

fix mask eval (#498)

上级 da4bd759
...@@ -121,6 +121,7 @@ def mask_eval(results, anno_file, outfile, resolution, thresh_binarize=0.5): ...@@ -121,6 +121,7 @@ def mask_eval(results, anno_file, outfile, resolution, thresh_binarize=0.5):
im_id = int(im_ids[i][0]) im_id = int(im_ids[i][0])
clsid_scores = bboxes[s:s + num][:, 0:2] clsid_scores = bboxes[s:s + num][:, 0:2]
mask = masks[s:s + num] mask = masks[s:s + num]
s += num
for j in range(num): for j in range(num):
clsid, score = clsid_scores[j].tolist() clsid, score = clsid_scores[j].tolist()
catid = int(clsid2catid[clsid]) catid = int(clsid2catid[clsid])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册