Skip to content

youthM/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

About

Spatial Transformer Networks

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 79.4%
  • Python 8.1%
  • Cuda 6.3%
  • CMake 2.7%
  • Protocol Buffer 1.6%
  • MATLAB 0.9%
  • Other 1.0%