cmd.sh

#!/usr/bin/env bash

# model version 2
python -u train.py --version 2 --batch-size 28 --optim adam --lr 1e-4 --lr-decay 0.1 --steps 30,50 --epoches 70 --gpu 0,1 --backbone resnet50 --freq 50 --prefix test  --num-stage 3 --num-channel 256

# model version 3
python -u train.py --version 3 --batch-size 18 --optim adam --lr 1e-4 --lr-decay 0.1 --steps 30,50 --epoches 70 --gpu 0,1 --backbone resnet50 --freq 100 --prefix test --num-channel 256

# model detection
python -u train_det.py --batch-size 32 --optim sgd --lr 1e-3 --lr-decay 0.1 --steps 5,10 --epoches 15 --gpu 0,1 --backbone resnet50 --freq 50 --prefix test

# prediction
python -u predict.py --model ./output/V2.test-vgg19-S3-C64-BS32-sgd-0001.params --version 2 --gpu 0 --num-worker 3 --multi-scale --type val --show

# detection prediction
python -u predict_det.py --det-model ./output/Det.default-resnet50-BS32-sgd-0015.params --kps-model ./output/V2.test-vgg19-S3-C64-BS32-sgd-0082.params --version 2 --gpu 0 --num-worker 3 --multi-scale --type val --show