File tree 1 file changed +37
-0
lines changed
1 file changed +37
-0
lines changed Original file line number Diff line number Diff line change
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
+ " $@ "
You can’t perform that action at this time.
0 commit comments