Skip to content
Implementation of our paper - Recurrent Convolutional Network for Video-based Person Re-Identification
Branch: master
Clone or download
niallmcl Update
Updated number of training epochs needed to reproduce results in paper
Latest commit 331cfae Jan 11, 2018
Type Name Latest commit message Commit time
Failed to load latest commit information.
.gitattributes 🎪 Added .gitattributes & .gitignore files Aug 8, 2016
.gitignore 🎪 Added .gitattributes & .gitignore files Aug 8, 2016
buildModel.lua Formatting Aug 8, 2016
computeOpticalFlow.m Bug fix Aug 8, 2016
prepareDataset.lua Formatting Aug 8, 2016 Update Jan 10, 2018
test.lua Formatting Aug 8, 2016
train.lua Formatting Aug 8, 2016

##Recurrent Convolutional Network for Video-based Person Re-Identification

Code for our CVPR 2016 paper that performs video re-identification.

If you use this code please cite:

  	title={Recurrent Convolutional Network for Video-based Person Re-Identification},
  	author={McLaughlin, N and Martinez del Rincon, J and Miller, P},

##Summary We perform video re-identification by taking a sequence of images and training a neural network to produce a single feature that represents the whole sequence. The feature vectors for different sequences can be compared using Euclidean distance. A smaller Euclidean distance indicates increased similarity between sequences. The sequence feature is produced using temporal pooling which averages the network's response at all time-steps in the sequence.

##Information A slightly cleaned up implementation of our video re-id system is provided here. If possible I will clean-up and improve the code in future.

This code is capable of training a video re-identification network on the iLids video or PRID datasets and saving the learned network for later use. The saved network parameters can be loaded from disk and used to run the evaluation code without needing to train the network again.

The optical flow files were produced using the Matlab code in computeOpticalFlow.m

This matlab code should be used to generate optical flow files before training the neural network. Alternatively, use the flag –dissableOpticalFlow

NOTE - Modify lines 70-77 of videoReid.lua to point to the directories containing the video-reid datasets and generated optical flow files

##Running the code

For this code to run you must have Torch7 installed with the nn, nnx, cunn, rnn, image, optim and cutorch pacakges.

You must have an Nvidia GPU in order to use CUDA. See for details.

Example command-line options that will allow you to run the code in standard configuration (Update 10 Jan 2018 - please use use 1000 epochs for training in order to reproduce results from the paper):

th videoReid.lua -nEpochs 1000 -dataset 1 -dropoutFrac 0.6 -sampleSeqLength 16 -samplingEpochs 100 -seed 1
You can’t perform that action at this time.