Commit 022b25cd authored by Sylvus's avatar Sylvus Committed by drpngx
Browse files

Fix for the IOU metric (#12709)

* Fixed mean iou case when a class does not appear in the labels nor in the prediction.

* Added 3 tests for both mean_iou and streaming_mean_iou, 2 of which would fail with the previous code and one to make sure the behavior is still correct in the normal case. Fixed broken tests as well.

* Added check for div by 0 in iou metric.

* Add space around operator

As per style guide.
parent 10c871ed
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment