UNet for non-square input images?

[Attila94]

When I train a model with square input size, all goes well. However, when I train a model with an input layer of, for example, 512x768, the training loss increases to up to 97% but the output is almost completely empty.

Anybody succeeded using UNet with a non-square input size?

[guanzzz]

yes, you can. But you need to know that the u net output is actually probability rather than pixel value. So what I did was multiply a constant and take threshold to get the segmentation result

[Attila94]

Turns out it was a matter of unbalanced data, had nothing to do with the input ratio.

[Mahi-Mai]

Hi @Attila94

Are you also making predictions on non-square images? This is something I'm struggling with right now. If I want to make a prediction, I need to segment my image into square subsamples, and I don't like that.

[ICG14]

Could you solve it? I am struggling to train a UNET CNN with rectangular images 512x896

[Attila94]

I'm not sure what the issue is but it has probably nothing to do with the input shape - UNet works for all input sizes.