File tree Expand file tree Collapse file tree 1 file changed +37
-0
lines changed Expand file tree Collapse file tree 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