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 dla_3x --dataset coco_hp --batch_size 128 --master_batch 9 --lr 5e-4 --load_model ../models/ctdet_coco_dla_2x.pth --gpus 0,1,2,3,4,5,6,7 --num_workers 16 --num_epochs 320 lr_step 270,300
|
2019-04-17 22:53:59 +08:00
|
|
|
# or use the following command if your have dla_1x trained
|
2019-04-18 06:04:36 +08:00
|
|
|
# python main.py multi_pose --exp_id dla_3x --dataset coco_hp --batch_size 128 --master_batch 9 --lr 5e-4 --gpus 0,1,2,3,4,5,6,7 --num_workers 16 --load_model ../exp/multi_pose/dla_1x/model_90.pth --resume
|
2019-04-17 22:53:59 +08:00
|
|
|
# test
|
2019-04-18 06:04:36 +08:00
|
|
|
python test.py multi_pose --exp_id dla_3x --dataset coco_hp --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 dla_3x --dataset coco_hp --keep_res --resume --flip_test
|
2019-04-17 22:53:59 +08:00
|
|
|
cd ..
|