faster-rcnn

This is an experimental Torch7 implementation of Faster RCNN - a convnet for object detection with a region proposal network. For details about R-CNN please refer to the paper Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks by Shaoqing Ren, Kaiming He, Ross Girshick, Jian Sun.

Work in progress

Status: Basic detection in my personal environment works. A 'small' network is used that can be trained on a 4 GB GPU with 800x450 images. Began experimenting with ImageNet: create-imagenet-traindat.lua can be used to create a training data file for the ILSVRC2015 dataset.

Todo:

• [!] regularly evaluate net during traning to compute test-set loss
• generate training graph with gnuplot
• add final per class non-maximum suppression to generate final proposals (already included but eval code rewrite still pending)
• remove hard coded path, create full set of command line options
• add parameters to separately enable/disable training of bounding box proposal-network and fine-tuning + classification.

Experiments to run:

• test smaller networks
• 6x6 vs. 7x7 classification ROI-pooling output size
• impact of RGB, YUV, Lab color space
• test relevance of local contrast normalization

References / Review / Useful Links

• SPP Paper
• Fast R-CNN paper
• R-CNN paper
• vgg net, cifar.torch
• 55 epoche learn rate schedule