Hi @ptrblck
I’m not sure I follow where to drop the channels. Is it at the time of generating the mask? and have the mask generated as [batch, width, height]?
If so, how should I encode and generate the mask?