Skip to content

Commit 7172d80

Browse files
committed
Add dukeMTMC-ReID train bash script
1 parent b00f733 commit 7172d80

File tree

1 file changed

+37
-0
lines changed

1 file changed

+37
-0
lines changed

duke_mtmc_train.sh

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,37 @@
1+
#!/bin/sh
2+
#
3+
# This file calls train.py with all hyperparameters as for the TriNet
4+
# experiment on DukeMTMC-ReID in the original paper.
5+
6+
if [ "$#" -lt 3 ]; then
7+
echo "Usage: $0 PATH_TO_IMAGES RESNET_CHECKPOINT_FILE EXPERIMENT_ROOT ..."
8+
echo "See the README for more info"
9+
echo "Download ResNet-50 checkpoint from https://github.com/tensorflow/models/tree/master/research/slim#pre-trained-models"
10+
exit 1
11+
fi
12+
13+
# Shift the arguments so that we can just forward the remainder.
14+
IMAGE_ROOT=$1 ; shift
15+
INIT_CHECKPT=$1 ; shift
16+
EXP_ROOT=$1 ; shift
17+
18+
python train.py \
19+
--train_set data/dukeMTMC_train.csv \
20+
--model_name resnet_v1_50 \
21+
--image_root $IMAGE_ROOT \
22+
--initial_checkpoint $INIT_CHECKPT \
23+
--experiment_root $EXP_ROOT \
24+
--flip_augment \
25+
--crop_augment \
26+
--embedding_dim 128 \
27+
--batch_p 18 \
28+
--batch_k 4 \
29+
--pre_crop_height 288 --pre_crop_width 144 \
30+
--net_input_height 256 --net_input_width 128 \
31+
--margin soft \
32+
--metric euclidean \
33+
--loss batch_hard \
34+
--learning_rate 3e-4 \
35+
--train_iterations 25000 \
36+
--decay_start_iteration 15000 \
37+
"$@"

0 commit comments

Comments
 (0)