ptrblck:

Is the original paper also returning the output in the original shape?


Yes. The original implementation, in the test case, returns the color image at approximately the original size. It uses scipy.ndimage.zoom () (https://docs.scipy.org/doc/scipy/reference/generated/scipy.ndimage.zoom.html 2) to increase the image size. I tried it here, but it didn’t work, the image was blurry.



 ptrblck:

If so, could you explain their architecture a bit or link the paper here?


I’ll send you the article link: https://arxiv.org/pdf/1603.08511.pdf
In the page 24 is the architecture configuration.