#!/bin/bash
set -x set -e
export PYTHONUNBUFFERED="True"
GPU_ID=$1 DATASET=$2 NET=$3
array=( $@ ) len=${#array[@]} EXTRA_ARGS=${array[@]:3:$len} EXTRA_ARGS_SLUG=${EXTRA_ARGS// /_}
case ${DATASET} in pascal_voc) TRAIN_IMDB="voc_2007_trainval" TEST_IMDB="voc_2007_test" #STEPSIZE="[50000]" #ITERS=70000 STEPSIZE="[500]" ITERS=10000 ANCHORS="[8,16,32]" RATIOS="[0.5,1,2]" ;; pascal_voc_0712) TRAIN_IMDB="voc_2007_trainval+voc_2012_trainval" TEST_IMDB="voc_2007_test" #STEPSIZE="[80000]" #ITERS=110000 STEPSIZE="[200]" ITERS=3200 ANCHORS="[8,16,32]" RATIOS="[0.5,1,2]" ;; coco) TRAIN_IMDB="coco_2014_train+coco_2014_valminusminival" TEST_IMDB="coco_2014_minival" #STEPSIZE="[350000]" #ITERS=490000 STEPSIZE="[100]" ITERS=4000 ANCHORS="[4,8,16,32]" RATIOS="[0.5,1,2]" ;; *) echo "No dataset given" exit ;; esac
LOG="experiments/logs/${NET}_${TRAIN_IMDB}_${EXTRA_ARGS_SLUG}_${NET}.txt.`date +'%Y-%m-%d_%H-%M-%S'`" exec &> >(tee -a "$LOG") echo Logging output to "$LOG"
set +x if [[ ! -z ${EXTRA_ARGS_SLUG} ]]; then NET_FINAL=output/${NET}/${TRAIN_IMDB}/${EXTRA_ARGS_SLUG}/${NET}_faster_rcnn_iter_${ITERS}.ckpt else NET_FINAL=output/${NET}/${TRAIN_IMDB}/default/${NET}_faster_rcnn_iter_${ITERS}.ckpt fi set -x
if [ ! -f ${NET_FINAL}.index ]; then if [[ ! -z ${EXTRA_ARGS_SLUG} ]]; then CUDA_VISIBLE_DEVICES=${GPU_ID} time python ./tools/trainval_net.py \ --weight data/imagenet_weights/${NET}.ckpt \ --imdb ${TRAIN_IMDB} \ --imdbval ${TEST_IMDB} \ --iters ${ITERS} \ --cfg experiments/cfgs/${NET}.yml \ --tag ${EXTRA_ARGS_SLUG} \ --net ${NET} \ --set ANCHOR_SCALES ${ANCHORS} ANCHOR_RATIOS ${RATIOS} \ TRAIN.STEPSIZE ${STEPSIZE} ${EXTRA_ARGS} else CUDA_VISIBLE_DEVICES=${GPU_ID} time python ./tools/trainval_net.py \ --weight data/imagenet_weights/${NET}.ckpt \ --imdb ${TRAIN_IMDB} \ --imdbval ${TEST_IMDB} \ --iters ${ITERS} \ --cfg experiments/cfgs/${NET}.yml \ --net ${NET} \ --set ANCHOR_SCALES ${ANCHORS} ANCHOR_RATIOS ${RATIOS} \ TRAIN.STEPSIZE ${STEPSIZE} ${EXTRA_ARGS} fi fi
./experiments/scripts/test_faster_rcnn.sh $@
扫码关注我们
微信号：SRE实战
拒绝背锅 运筹帷幄