Tramac / Fast-SCNN-pytorch

A PyTorch Implementation of Fast-SCNN: Fast Semantic Segmentation Network

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Wrong calculation of IoU

JonasFrey96 opened this issue · comments

Dear Tramac,
Thanks for your work at first!
The way you calculate the MeanIOU is incorrect. From my understanding you at first have to calculate the Intersection of Union for each class that occures in the target image. Then you have to take the mean. What you do is summing up all pixels that fall inside the union or the intersection and simply take the average. This leads to an incorrect metric from my understand. Please correct me if I am wrong. It is pretty clearly stated here averaged over all classes: \url{https://www.jeremyjordan.me/evaluating-image-segmentation-models/}

done