Hey - it doesn't make sense to train segnet on a single class. In this situation, segnet will simply learn to predict every pixel as this class, and you will see loss=0.
ArunJ1 opened this issue · comments
Hey - it doesn't make sense to train segnet on a single class. In this situation, segnet will simply learn to predict every pixel as this class, and you will see loss=0.
Instead, you need to have a background class that isn't ignored. It is essential to have a background class to learn what isn't your object of interest.
Therefore you should set num_output: 2
, label your object and background classes as 0 and 1 and have no ignore label.
Cheers.
Originally posted by @alexgkendall in #31 (comment)
Thanks for your advice @alexgkendall ....I have found this after searching around for long time....Thanks Again..
Pls could you confirm whether it is mandatory to calculate the class weighting and add??
As mentioned above, I have only single class to predict. That's why i would like to confirm whether it is mandatory.
Thanks in advance