oarriaga / STN.keras

Implementation of spatial transformer networks (STNs) in keras 2 with tensorflow as backend.
MIT License
278 stars 75 forks source link

Wrong output on inputting image of different width and height #15

Open ruqayya opened 4 years ago

ruqayya commented 4 years ago

Hi,

The transformed image (output of STN) is correct when the width and height of the input image are the same. On inputting an image of size 512x256, the transformed image was duplicated twice. However, it seems to work fine for an input image of size 256x512. Below is the output of STN for an image of size 512x256 on applying transformation to a fixed image. The expected output should be similar to the moving image.

image