CenterNet/experiments/multi_pose_hg_1x.sh

9 lines
452 B
Bash
Raw Permalink Normal View History

2019-04-17 22:53:59 +08:00
cd src
# train
2019-04-18 06:04:36 +08:00
python main.py multi_pose --exp_id hg_1x --dataset coco_hp --arch hourglass --batch_size 24 --master_batch 4 --lr 2.5e-4 --load_model ../models/ctdet_coco_hg.pth --gpus 0,1,2,3,4 --num_epochs 50 --lr_step 40
2019-04-17 22:53:59 +08:00
# test
2019-04-18 06:04:36 +08:00
python test.py multi_pose --exp_id hg_1x --dataset coco_hp --arch hourglass --keep_res --resume
2019-04-17 22:53:59 +08:00
# flip test
2019-04-18 06:04:36 +08:00
python test.py multi_pose --exp_id hg_1x --dataset coco_hp --arch hourglass --keep_res --resume --flip_test
2019-04-17 22:53:59 +08:00
cd ..