The repository contains the training and evaluation code for Siamese Network https://en.wikipedia.org/wiki/Siamese_neural_network on MARS Dataset (Zheng L, Bie Z, Sun Y, Wang J, Su C, Wang S, Tian Q. Mars: A video benchmark for large-scale person re-identification. InEuropean Conference on Computer Vision 2016 Oct 8 (pp. 868-884). Springer, Cham).
Based on Tensorflow==1.14
Training instructions:
python3 train_net.py --help usage: train_net.py [-h] [-p [PATH]] [--pre_trained PRE_TRAINED] [-b BATCH_SIZE] [--epochs EPOCHS]
Training options.
optional arguments:
-h, --help show this help message and exit
-p [PATH], --path [PATH]
Path to Training data
--pre_trained PRE_TRAINED
Use pretrained ResNet50 model from TensorNets? False
by Default
-b BATCH_SIZE, --batch_size BATCH_SIZE
Batch size to use
--epochs EPOCHS number of epochs to train the model for