-
Notifications
You must be signed in to change notification settings - Fork 5
/
run_local.sh
14 lines (13 loc) · 930 Bytes
/
run_local.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
PRETRAIN_MODEL='./model_best.pth.tar'
VOS_SAVE_RESULT_DIR='/data01/jiaxu/saved_model/stage1'
INT_SAVE_RESULT_DIR='/data01/jiaxu/saved_model/stage2'
INT_RESULT_DIR='/data01/jiaxu/saved_model/result'
RESCALE=416
RANDOMCROP=416
DATA_ROOT='/yourdataroot/DAVIS'
echo 'Stage1 training'
python train_stage1.py --SAVE_RESULT_DIR $VOS_SAVE_RESULT_DIR --PRETRAINED_MODEL $PRETRAIN_MODEL --DATA_ROOT $DATA_ROOT --TRAIN_BATCH_SIZE 2 --DATA_RESCALE $RESCALE --DATA_RANDOMCROP $RANDOMCROP --TRAIN_LR 0.0007 --MODEL_MAX_LOCAL_DISTANCE 12
echo 'Stage2 training'
python train_stage2.py --SAVE_RESULT_DIR $INT_SAVE_RESULT_DIR --SAVE_VOS_RESULT_DIR $VOS_SAVE_RESULT_DIR --DATA_ROOT $DATA_ROOT --DATA_RESCALE $RESCALE --DATA_RANDOMCROP $RANDOMCROP --PRETRAINED_MODEL $PRETRAIN_MODEL
echo 'Testing'
python test.py --DATA_ROOT $DATA_ROOT --SAVE_RESULT_DIR $INT_SAVE_RESULT_DIR --RESULT_ROOT $INT_RESULT_DIR --MODEL_USEIntSeg False --TEST_MODE True