prediction error about pytorch-unet

[milesial]

Hi, i think it has to to with the channels of the images. They should be loaded as RGB, so they have 3 channels, I don't know why your images have 4 channels (maybe the alpha?)?

[dingguo1996]

hey I get the same problem as yours

[dingguo1996]

my problem is "invalid argument 0: Sizes of tensors must match except in dimension 1. Got 52 and 51 in dimension 2",when I try to run the prediction.py

[dingguo1996]

i found the way to solve my problem in the closed issue,thx!