applicable to multichannel mask (multiclass) predictions ? meaning each class is trained on different channel

[amardeepjaiman]

Can this script be used to innference multiclass on multiple channels ? We have multichannel masks prepared while training . Please suggest.

[bnsreenu]

Yes, this can be used for multichannel predictions, just define nb_classes = 6 (in my case I have 6 classes).