-
Notifications
You must be signed in to change notification settings - Fork 302
add VOC instance segmentation evaluation #554
Conversation
d37b7ec
to
b89995a
Compare
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.
Great!
'voc_instance_segmentation_result_test_2018_04_04.npz'))[0], | ||
encoding='latin1') | ||
|
||
def test_eval_detection_voc(self): |
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.
test_eval_instance_segmentation_voc
@classmethod | ||
def setUpClass(cls): | ||
base_url = 'https://github.com/knorth55/' \ | ||
'share-weights/releases/download/v0.0.0' |
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.
I added you to a maintainer of yuyu2172/share-weights
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.
thx!
dataset. Each element of :obj:`pred_masks` is an object mask | ||
and is an array whose shape is :math:`(R, H, W)`, | ||
where :math:`R` corresponds | ||
to the number of masks, which may vary among imagse. |
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.
imagse -> images
|
||
pred_labels = iter(pred_labels) | ||
pred_scores = iter(pred_scores) | ||
gt_labels = iter(gt_labels) |
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.
can you iter
all inputs?
dataset. Each element of :obj:`pred_masks` is an object mask | ||
and is an array whose shape is :math:`(R, H, W)`, | ||
where :math:`R` corresponds | ||
to the number of masks, which may vary among imagse. |
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.
imagse --> images
length is :math:`N`. | ||
iou_thresh (float): A prediction is correct if its Intersection over | ||
Union with the ground truth is above this value. | ||
use_07_metric (bool): Whether to use PASCAL VOC 2007 evaluation metric |
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.
I think the default value should be False to be consistent with eval_detection_voc
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.
OK, I set as default False
selec[gt_idx] = True | ||
else: | ||
match[l].append(0) | ||
|
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.
I think it is same to insert error check like eval_detection_voc
https://github.com/chainer/chainercv/blob/master/chainercv/evaluations/eval_detection_voc.py#L214
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.
I updated
LGTM |
eval_instance_segmentation_voc
test_eval_instance_segmentation_voc
eval_instance_segmentation_voc