Skip to content
Snippets Groups Projects
Commit b96d2e03 authored by Thang Vu's avatar Thang Vu Committed by Kai Chen
Browse files

fix mAP w.r.t. object scales (#1679)

parent 2b89634f
No related branches found
No related tags found
No related merge requests found
...@@ -188,6 +188,7 @@ def segm2json(dataset, results): ...@@ -188,6 +188,7 @@ def segm2json(dataset, results):
for i in range(bboxes.shape[0]): for i in range(bboxes.shape[0]):
data = dict() data = dict()
data['image_id'] = img_id data['image_id'] = img_id
data['bbox'] = xyxy2xywh(bboxes[i])
data['score'] = float(mask_score[i]) data['score'] = float(mask_score[i])
data['category_id'] = dataset.cat_ids[label] data['category_id'] = dataset.cat_ids[label]
if isinstance(segms[i]['counts'], bytes): if isinstance(segms[i]['counts'], bytes):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment