search

skaae / transformer_network

Spatial Transformer network
MIT License
190 stars 52 forks source link

add to Keras #1

Closed pranv closed 9 years ago

skaae commented 9 years ago

I think you should open this as a Keras issue? Feel free to copy the code as long as you write where you got i from :)

pranv commented 9 years ago

Sure! I just thought that you would be able to do it faster (maybe less than an hour) than me ( studying code and transferring

skaae commented 9 years ago

yes. I'll open an issue in Keras, if there is interested i'll write the layer. It might also be good to have some code review + tests.

EderSantana commented 9 years ago

Hi, I copy pasted @skaae 's lasagne layer to a keras layer (with proper citation, but feel free to add your name or whatever you prefer to it). An example usage is here: https://github.com/EderSantana/seya/blob/master/examples/Spatial%20Transformer%20Networks.ipynb

But I tried several architectures for the localization networks (locnet in the code) but I'm not sure results look correct. If anybody is interested pls check it out and let me know what you found out.

Victoria2333 commented 7 years ago

@skaae Does this code have tensorflow version?? Thank you~~~