Skip to content
Go to file

Latest commit


Git stats


Failed to load latest commit information.
Latest commit message
Commit time

This project prvides the Torch solution for the paper Recurrent Convolutional Neural Network for object recognition. The code is heavily inspired by facebook/fb.resnet.torch.


  1. A GPU machine with Torch and its cudnn bindings. See Installing Torch.

  2. Download Torch version cifar10 and svhn datasets, and put them to rcnn/data/cifar/ and rcnn/data/svhn/, respectively.

#How to use Run main.lua with options to train RCNN models.

An cifar10 example with error rate 4.59% is:

CUDA_VISIBLE_DEVICES=0,1 th main.lua -dataset cifar10 -model rcl3_large -nGPU 2 -nThreads 4 -lr 0.1 -nChunks 100 -batchSize 64

An svhn example with error rate 1.5% is:

CUDA_VISIBLE_DEVICES=0,1,2,3 th main.lua -dataset svhn -model rcl3 -nGPU 4 -nThreads 8 -lr 0.1 -nChunks 100 -batchSize 256

To see all options and their default value, run:

th main.lua -help

#Code introduction

  1. main.lua: Overall procedure to run the code.

  2. dataset.lua: Prepare mini-batchs from specified datasets, including possible data augmentation.

  3. data.lua: Initiate the dataset and setup multi-thread data loaders.

  4. model.lua: Initiate the network models. Model files are placed in rcnn/models/.

  5. train.lua: Train and test network models.

  6. parse.lua: Parse the input options.


No description, website, or topics provided.



No releases published


You can’t perform that action at this time.