From d5a5fa4b30fcdd15cff1a47b53cf96296b15cbf5 Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Tue, 14 Apr 2020 16:04:43 +0800 Subject: [PATCH] fix mask eval (#498) --- ppdet/utils/coco_eval.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ppdet/utils/coco_eval.py b/ppdet/utils/coco_eval.py index fa86c292a..d2ceeacb4 100644 --- a/ppdet/utils/coco_eval.py +++ b/ppdet/utils/coco_eval.py @@ -121,6 +121,7 @@ def mask_eval(results, anno_file, outfile, resolution, thresh_binarize=0.5): im_id = int(im_ids[i][0]) clsid_scores = bboxes[s:s + num][:, 0:2] mask = masks[s:s + num] + s += num for j in range(num): clsid, score = clsid_scores[j].tolist() catid = int(clsid2catid[clsid]) -- GitLab