test: Fix seg_iou in case NaN

bit-bots / YOEO

YouOnlyEncodeOnce - A CNN for Embedded Object Detection and Semantic Segmentation

GNU General Public License v3.0

22 stars 4 forks source link

test: Fix seg_iou in case NaN #66

Closed jaagut closed 1 year ago

jaagut commented 1 year ago

Proposed changes

Fixes an issue, where the segmentation IOU can be NaN, if a whole batch does not contain any pixels of a segmentation class. Removes cases, where the segmentation IOU is NaN.

Related issues

Necessary checks

[x] Update poetry package version semantically
[ ] Write documentation
[ ] Create issues for future work
[ ] Test on your machine

jaagut commented 1 year ago

Please test this yourself