Skip to content

christopher5106/last_caffe_with_stn

Repository files navigation

Spatial Transformer Networks and LSTM RNN in Caffe

Spatial Transformer Networks

Spatial Transformer Networks from https://github.com/daerduoCarey/SpatialTransformerLayer with an updated (last) version of Caffe and some code cleaning and corrections.

Compile Caffe following my tutorial on iOS or Ubuntu.

My tutorial about Spatial Transformer Networks.

Commands to build a Docker :

docker build -f docker/standalone/cpu/Dockerfile  -t caffe/stn:v1 .

or pull it directly :

docker pull christopher5106/caffe:stn

LSTM RNN

LSTM from https://github.com/junhyukoh/caffe-lstm

My tutorial about LSTM in Caffe.

Repeat Layer

As in https://github.com/skaae/recurrent-spatial-transformer-code I created a RepeatLayer designed for Recurrent Neural Nets.

Smooth L1 layer

From Ross Girshick