-
Notifications
You must be signed in to change notification settings - Fork 302
Fix SegNet example's evaluation protocol #249
Conversation
examples/segnet/README.md
Outdated
``` | ||
|
||
e.g., | ||
|
||
``` | ||
python evaluate.py 0 result/model_iteration-16000 | ||
python evaluate.py --gpu 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.
This example is no longer needed.
mean_accs.append(n_true_positive[cls_i] / n_true[cls_i]) | ||
pixel_accs.append([n_true_positive[cls_i], n_true[cls_i]]) | ||
imgs, pred_values, gt_values = apply_prediction_to_iterator( | ||
model.predict, it) |
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 delete imgs
to save the memory. https://github.com/chainer/chainercv/blob/master/examples/detection/eval_voc07.py#L81-L82
examples/segnet/evaluate.py
Outdated
gt_labels, = gt_values | ||
|
||
confusion = calc_semantic_segmentation_confusion( | ||
pred_labels, gt_labels, len(camvid_label_names)) |
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.
len(camvid_label_names)
is no longer needed.
https://github.com/chainer/chainercv/blob/master/chainercv/evaluations/eval_semantic_segmentation_iou.py#L7
830b224
to
8c8ace9
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.
LGTM
Split contents in #238.