Variable classes about mobile-semantic-segmentation HOT 2 CLOSED

I guess the other option would be to convert my images - which are (1, h, w), with pixels values representing the class number - to (num_classes, h, w), with each channel representing a single one-hot coded class... So the last activation layer should output (batches, num_classes, h, w)... (though the model summary prints this as (None, h, w, num_classes)).

NOTE: I would think that multi-class should also probably use softmax instead of sigmoid...

from mobile-semantic-segmentation.

Yes. The output shape will be something like (n_batch, h, w, n_classes). Then, apply softmax loss across last dimension.

If I have time, I will add example to segment hair, face and background.

from mobile-semantic-segmentation.