Solved by KFrank in post #2 
              
              
                Hi William! 

As an aside, it sounds like your current CNN performs binary (i.e., 
background-foreground) segmentation. 

It sounds like your desired use case is multi-class (with NB_CLASS 
classes) semantic segmentation. 

First, once you’re down to (BATCH_SIZE, 1, HEIGHT, WIDTH), that 
is, a singl…