-
Notifications
You must be signed in to change notification settings - Fork 306
Add COCO instance segmentation evaluation #671
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please add doc link
.. csv-table:: | ||
:header: name, shape, dtype, format | ||
|
||
:obj:`pred_masks`, ":math:`[(R, H, W)]`", :obj:`float32`, \ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
bool
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
format is wrong
":math:`[0, \#fg\_class - 1]`" | ||
:obj:`pred_scores`, ":math:`[(R,)]`", :obj:`float32`, \ | ||
-- | ||
:obj:`gt_masks`, ":math:`[(R, H, W)]`", :obj:`float32`, \ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
ditto
|
||
This function evaluates predicted bounding boxes obtained from a dataset | ||
by using average precision for each class. | ||
The code is based on the evaluation code used in MS COCO. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
bounding boxes ---> instance segmentation
def eval_instance_segmentation_coco( | ||
pred_masks, pred_labels, pred_scores, | ||
gt_masks, gt_labels, gt_areas=None, gt_crowdeds=None): | ||
"""Evaluate detections based on evaluation code of MS COCO. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
detection --> instance segmentation
:header: key, type, description | ||
|
||
ap/iou=0.50:0.95/area=all/max_dets=100, *numpy.ndarray*, \ | ||
[#coco_det_eval_1]_ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
change coco_det_eval_1
to coco_inst_segm_eval_1
.
The link conflicts with eval_detection_coco
.
You will notice the problem after compiling.
Thank you for review. I updated. |
merge after #665
This PR is based on https://github.com/yuyu2172/coco-evaluation