T-CNN: Tubelets with Convolution Neural Networks

Introduction

The TCNN framework is a deep learning framework for object detection in videos. This framework was orginally designed for the ImageNet VID chellenge in ILSVRC2015.

Citing T-CNN

If you are using the T-CNN code in you project, please cite the following works.

@inproceedings{kang2016object,
  Title = {Object Detection from Video Tubelets with Convolutional Neural Networks},
  Author = {Kang, Kai and Ouyang, Wanli and Li, Hongsheng and Wang, Xiaogang},
  Booktitle = {CVPR},
  Year = {2016}
}
@article{kang2016tcnn,
  title={T-CNN: Tubelets with Convolutional Neural Networks for Object Detection from Videos},
  author={Kang, Kai and Li, Hongsheng and Yan, Junjie and Zeng, Xingyu and Yang, Bin and Xiao, Tong and Zhang, Cong and Wang, Zhe and Wang, Ruohui and Wang, Xiaogang and Ouyang, Wanli},
  journal={arXiv preprint},
  year={2016}
}

License

T-CNN is released under the MIT License.

ImageNet 2015 VID detection results

Track	Validation Set	Test Set	Rank in ILSVRC2015
Provided	73.8	67.8	#1
Additional	77.0	69.7	#2

Installations

Prerequisites

caffe with Python layer and pycaffe
GNU Parallel
matutils
FCN tracker
Matlab with python engine

Instructions

Clone the repository and sub-repositories from GitHub, let $TCNN_ROOT represents the root directory of the repository.

    $ # clone the repository
    $ git clone --recursive https://github.com/myfavouritekk/T-CNN.git
    $ cd $TCNN_ROOT
    $ # checkout the ImageNet 2015 VID branch
    $ git checkout ilsvrc2015vid

Compilation for vdetlib

    $ cd $TCNN_ROOT/vdetlib
    $ make
    $ export PYTHONPATH=$TCNN_ROOT/vdetlib:$PYTHONPATH

Download and install caffe in the External directory

    $ git clone https://github.com/BVLC/caffe.git External/caffe
    $ # modify `Makefile.config` and build with Python layer and pycaffe
    $ # detailed instruction, please follow http://caffe.berkeleyvision.org/installation.html
    $ export PYTHONPATH=$TCNN_ROOT/External/caffe/python:$PYTHONPATH

Download a modified version of FCN Tracker originally developed by Lijun Wang et. al.

    $ git clone --recursive -b T-CNN https://github.com/myfavouritekk/FCNT External/fcn_tracker_matlab
    $ # compile the caffe-fcn_tracking and configure FCNT

Demo

Extract the sample data and still-image detection results

    $ cd $TCNN_ROOT
    $ unzip sample_data.zip -d data/

Generate optical flow for the videos

    $ mkdir ./data/opt_flow
    $ ls ./data/frames |
        parallel python tools/data_proc/gen_optical_flow.py ./data/frames/{} ./data/opt_flow/{} --merge

Multi-context suppression and motion-guided propagation in Matlab

    >> addpath(genpath('tools/mcs_mgp'));
    >> mcs_mgp('data/opt_flow', 'data/scores', 'data/mcs_mgp')

Tubelet tracking and re-scoring

    $ # generate .vid protocol files
    $ ls data/frames | parallel python vdetlib/tools/gen_vid_proto_file.py {} $PWD/data/frames/{} data/vids/{}.vid
    $ # tracking from raw detection files
    $ find data/vids -type f -name *.vid | parallel -j1 python tools/tracking/greedy_tracking_from_raw_dets.py {} data/mcs_mgp/window_size_7_time_step_1_top_ratio_0.000300_top_bonus_0.400000_optflow/{/.} data/tracks/{/.} --thres 3.15 --max_frames 100 --num 30
    $ # spatial max-pooling
    $ find data/vids -type f | parallel python tools/scoring/tubelet_raw_dets_max_pooling.py {} data/tracks/{/.} data/mcs_mgp/window_size_7_time_step_1_top_ratio_0.000300_top_bonus_0.400000_optflow/{/.} data/score_proto/window_size_7_time_step_1_top_ratio_0.000300_top_bonus_0.400000_optflow_max_pooling/{/.} --overlap_thres 0.5

Tubelet visualization

    $ python tools/visual/show_score_proto.py data/vids/ILSVRC2015_val_00007011.vid data/score_proto/window_size_7_time_step_1_top_ratio_0.000300_top_bonus_0.400000_optflow_max_pooling/ILSVRC2015_val_00007011/ILSVRC2015_val_00007011.airplane.score

Beyond demo

Optical flow extraction

    $ python tools data_proc/gen_optical_flow.py -h

vdetlib for tracking and rescoring
Visualization tools in tools/visual.

Known Issues

Matlab engines may stall after long periods of tracking. Please consider to kill the certain matlab session to continue.

To-do list

Tubelet Bayesian classifier

sditeng/T-CNN