STN.keras
STN.keras copied to clipboard
oarriaga
Implementation of spatial transformer networks (STNs) in keras 2 with tensorflow as backend.
Traceback (most recent call last): File "train.py", line 16, in model = STN() File "/home/vision/STN/STN.py", line 29, in STN x = Conv2D(32, (3, 3), padding='same')(interpolated_image) File "/home/vision/.virtualenvs/dl4cv/lib/python3.5/site-packages/tensorflow_core/python/keras/engine/base_layer.py", line 812, in...
Hi, thank you for sharing your code. When I run it in a google colab (keras=2.4.3), I got following error. Any idea on how to correct it? Thank you. ---------------------------------------------------------------------------...
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...
Hello, I have tried your code for an image of size 48*48, i want to use the _transform function whose definition is like: _transform(self,X,affine_transformation,output_size) of class Bi-linear interpolation but i...
Hi, thanks for providing the code. I have run the code and the result worked out well ,but when I try to transform the code to recognize mnist of size...
