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 ..
|