Reshape

[mavaylon1]

What's the reasoning behind reshaping the masks and then the model output to (width*height, channel)?

When you run predictions you change back: pr = pr.reshape((output_height, output_width, n_classes)).argmax(axis=2).

Everywhere else I see (batch, height, width, n_classes) as the output and (batch, height, width, channels) as input