####This code is still under test and modification. If you are interested in the model, you may also visit (https://github.com/stupiding/rcnn), which implements an essentially same model for image classification.
The project aims to providing the Torch solution for the paper Convolutional Neural Networks with Intra-Layer Recurrent Connections for Scene Labeling. The code is modified from facebook/fb.resnet.torch.
#Requirements
-
A GPU machine with Torch and its cudnn bindings. See Installing Torch.
-
Download the siftflow dataset, tranform it to t7 format, and put it to scene-labeling/data/siftflow.
#How to use Run main.lua with options to train network models.
An example is:
CUDA_VISIBLE_DEVICES=0,1 th main.lua -dataset siftflow -model rcl3 -nGPU 2 -nThreads 4 -lr 0.1 -nChunks 100 -batchSize 64
To see all options and their default value, run:
th main.lua -help
#Code introduction
-
main.lua: Overall procedure to run the code.
-
dataset.lua: Prepare mini-batchs from specified datasets, including possible data augmentation.
-
data.lua: Initiate the dataset and setup multi-thread data loaders.
-
model.lua: Initiate the network models. Model files are placed in rcnn/models/.
-
train.lua: Train and test network models.
-
parse.lua: Parse the input options.