image scaling

[chenyili502]

In your project, you scaled the image, such as 256 * 256, but did you not scale back to the original image in the output. Please tell me if I understand correctly.

[chenyili502]

If the image is reshaped when it is input first, and there is no reshape when it is output from the network, is there a problem,