tracking-with-rnn
Videos and codebase of First step toward model-free, anonymous object tracking with recurrent networks.
Requirements
All models require Theano to train & test.
Additionally, ConvTracker requires keras.
Files
conv_base.py - ConvTracker
recurrent_base.py - RecTracker-ID
recurrent_att.py - RecTracker-Att-N. N can be specified in --grid_size option.
data_handler.py - Moving MNIST dataset generator
Usage
Run train.sh to train the RecTracker-Att-1 model.
Run test.sh to test the RecTracker-Att-1 model under different configurations.
RecTracker-ID and ConvTracker accept the same options.