Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
mmdet/core/evaluation/coco_utils.py
View file @ b96d2e03
...@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment