You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
#145
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.
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)
The text was updated successfully, but these errors were encountered: