In the test phase, how can I get the output image to be the same size as the original image instead of the image size of 256*256. Thanks!