Instead of binary mask If I use multiclass mask

[shiblyg]

will it be work if I use multiclass mask? Which part should I modify during training?

[zhangzc21]

I think you need to make some changes to the output layer of the auxiliary network.