/tf-spatial-transformer-networks

Spatial Transformer Networks in TensorFlow

Primary LanguagePython

TensorFlow implementation of STN

This is a TensorFlow implementation of the following paper:

Spatial Transformer Networks

The code can only run in the Eager Execution.

Results

Compared with a normal fully connected network:

STN outputs: