The Spatial Transformer Network [1] allows the spatial manipulation of data within the network.
API
A Spatial Transformer Network implemented in Tensorflow 0.7 and based on [2].
How to use
transformer(U, theta, out_size)
Parameters
U : float
The output of a convolutional net should have the
shape [num_batch, height, width, num_channels].
theta: float
The output of the
localisation network should be [num_batch, 6].
out_size: tuple of two ints
The size of the output of the network
Notes
To initialize the network to the identity transform init theta to :