CRFasRNN Training

[AlexTS1980]

How did you train CRFasRNN, specifially what did the last layer look like?

What I mean is that it couldn't be just a vector of 21 neurons, because you have to segment the object in addition to classification. Did you use a mask size the input image with 1's in the location of the object and 0 elsewhere? I see that the output is of shape (batch_size, num_class, width, height), but not sure how you used this for training.