diff --git a/experiments/ctdet_coco_hg104_cornernet.sh b/experiments/ctdet_coco_hg104_cornernet.sh index 7dbc072..a0718c2 100644 --- a/experiments/ctdet_coco_hg104_cornernet.sh +++ b/experiments/ctdet_coco_hg104_cornernet.sh @@ -1,13 +1,14 @@ -# ttrain +# train python src/main.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_cornernet --arch hourglass --batch_size 36 --master_batch 6 --lr 2.5e-4 --gpus 0,1,2,3 --num_workers 16 --coco_path $COCO_PATH --num_epochs 60 --lr_step 40,55 --load_model ./models/CornerNet_500000.pth +### for testing your own trainings, please remove '--model_v1' # test -python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_cornernet --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_cornernet.pth --coco_path $COCO_PATH +python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_cornernet --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_cornernet.pth --coco_path $COCO_PATH --model_v1 # flip test -python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_cornernet --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_cornernet.pth --coco_path $COCO_PATH --flip_test +python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_cornernet --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_cornernet.pth --coco_path $COCO_PATH --flip_test --model_v1 # multi scale test -python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_cornernet --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_cornernet.pth --coco_path $COCO_PATH --flip_test --test_scales 0.6,0.8,1,1.2,1.5,1.8 +python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_cornernet --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_cornernet.pth --coco_path $COCO_PATH --flip_test --test_scales 0.6,0.8,1,1.2,1.5,1.8 --model_v1 diff --git a/experiments/ctdet_coco_hg104_extremenet.sh b/experiments/ctdet_coco_hg104_extremenet.sh index 2576288..98df8f6 100644 --- a/experiments/ctdet_coco_hg104_extremenet.sh +++ b/experiments/ctdet_coco_hg104_extremenet.sh @@ -1,12 +1,14 @@ # train python src/main.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_extremenet --arch hourglass --batch_size 29 --master_batch 5 --lr 2.5e-4 --gpus 0,1,2,3 --num_workers 16 --coco_path $COCO_PATH --num_epochs 50 --lr_step 40 --load_model ./models/ExtremeNet_500000.pth + +### for testing your own trainings, please remove '--model_v1' # test -python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_extremenet --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_extremenet.pth --coco_path $COCO_PATH +python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_extremenet --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_extremenet.pth --coco_path $COCO_PATH --model_v1 # flip test -python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_extremenet --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_extremenet.pth --coco_path $COCO_PATH --flip_test +python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_extremenet --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_extremenet.pth --coco_path $COCO_PATH --flip_test --model_v1 # multi scale test -python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_extremenet --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_extremenet.pth --coco_path $COCO_PATH --flip_test --test_scales 0.6,0.8,1,1.2,1.5,1.8 +python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_extremenet --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_extremenet.pth --coco_path $COCO_PATH --flip_test --test_scales 0.6,0.8,1,1.2,1.5,1.8 --model_v1 diff --git a/experiments/ctdet_coco_hg104_scratch.sh b/experiments/ctdet_coco_hg104_scratch.sh index f90f1fb..e5e0b01 100644 --- a/experiments/ctdet_coco_hg104_scratch.sh +++ b/experiments/ctdet_coco_hg104_scratch.sh @@ -2,12 +2,13 @@ # train python src/main.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_scratch --arch hourglass --batch_size 36 --master_batch 6 --lr 2.5e-4 --gpus 0,1,2,3 --num_workers 16 --coco_path $COCO_PATH --num_epochs 100 --lr_step 90 +### for testing your own trainings, please remove '--model_v1' # test -python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_scratch --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_scratch.pth --coco_path $COCO_PATH +python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_scratch --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_scratch.pth --coco_path $COCO_PATH --model_v1 # flip test -python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_scratch --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_scratch.pth --coco_path $COCO_PATH --flip_test +python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_scratch --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_scratch.pth --coco_path $COCO_PATH --flip_test --model_v1 # multi scale test -python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_scratch --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_scratch.pth --coco_path $COCO_PATH --flip_test --test_scales 0.6,0.8,1,1.2,1.5,1.8 +python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_hg104_scratch --arch hourglass --keep_res --resume --load_model ./models/ctdet_coco_hg104_scratch.pth --coco_path $COCO_PATH --flip_test --test_scales 0.6,0.8,1,1.2,1.5,1.8 --model_v1 diff --git a/experiments/ctdet_coco_res101.sh b/experiments/ctdet_coco_res101.sh index 2227341..ed1f724 100644 --- a/experiments/ctdet_coco_res101.sh +++ b/experiments/ctdet_coco_res101.sh @@ -1,8 +1,10 @@ # train python src/main.py ctdet --houghnet --region_num 17 --vote_field_size 65 --exp_id coco_res_101 --arch res_101 --batch_size 44 --master_batch 8 --lr 1.75e-4 --gpus 0,1,2,3 --num_workers 16 --coco_path $COCO_PATH + +### for testing your own trainings, please remove '--model_v1' # test -python src/test.py ctdet --houghnet --region_num 17 --vote_field_size 65 --exp_id coco_res101 --arch res_101 --keep_res --resume --load_model ./models/ctdet_coco_res101.pth --coco_path $COCO_PATH +python src/test.py ctdet --houghnet --region_num 17 --vote_field_size 65 --exp_id coco_res101 --arch res_101 --keep_res --resume --load_model ./models/ctdet_coco_res101.pth --coco_path $COCO_PATH --model_v1 # flip test -python src/test.py ctdet --houghnet --region_num 17 --vote_field_size 65 --exp_id coco_res101 --arch res_101 --keep_res --resume --load_model ./models/ctdet_coco_res101.pth --coco_path $COCO_PATH --flip_test +python src/test.py ctdet --houghnet --region_num 17 --vote_field_size 65 --exp_id coco_res101 --arch res_101 --keep_res --resume --load_model ./models/ctdet_coco_res101.pth --coco_path $COCO_PATH --flip_test --model_v1 # multi scale test -python src/test.py ctdet --houghnet --region_num 17 --vote_field_size 65 --exp_id coco_res101 --arch res_101 --keep_res --resume --load_model ./models/ctdet_coco_res101.pth --coco_path $COCO_PATH --flip_test --test_scales 0.6,0.8,1,1.2,1.5,1.8 +python src/test.py ctdet --houghnet --region_num 17 --vote_field_size 65 --exp_id coco_res101 --arch res_101 --keep_res --resume --load_model ./models/ctdet_coco_res101.pth --coco_path $COCO_PATH --flip_test --test_scales 0.6,0.8,1,1.2,1.5,1.8 --model_v1 diff --git a/experiments/ctdet_coco_resdcn101.sh b/experiments/ctdet_coco_resdcn101.sh index 4cd51c5..6da267d 100644 --- a/experiments/ctdet_coco_resdcn101.sh +++ b/experiments/ctdet_coco_resdcn101.sh @@ -2,10 +2,12 @@ # Base Model # train python src/main.py ctdet --houghnet --region_num 17 --vote_field_size 65 --exp_id coco_resdcn_101 --arch resdcn_101 --batch_size 44 --master_batch 8 --lr 1.75e-4 --gpus 0,1,2,3 --num_workers 16 --coco_path $COCO_PATH + +### for testing your own trainings, please remove '--model_v1' # test -python src/test.py ctdet --houghnet --region_num 17 --vote_field_size 65 --exp_id coco_resdcn_101 --arch resdcn_101 --keep_res --resume --load_model ./models/ctdet_coco_resdcn101.pth --coco_path $COCO_PATH +python src/test.py ctdet --houghnet --region_num 17 --vote_field_size 65 --exp_id coco_resdcn_101 --arch resdcn_101 --keep_res --resume --load_model ./models/ctdet_coco_resdcn101.pth --coco_path $COCO_PATH --model_v1 # flip test -python src/test.py ctdet --houghnet --region_num 17 --vote_field_size 65 --exp_id coco_resdcn_101 --arch resdcn_101 --keep_res --resume --load_model ./models/ctdet_coco_resdcn101.pth --coco_path $COCO_PATH --flip_test +python src/test.py ctdet --houghnet --region_num 17 --vote_field_size 65 --exp_id coco_resdcn_101 --arch resdcn_101 --keep_res --resume --load_model ./models/ctdet_coco_resdcn101.pth --coco_path $COCO_PATH --flip_test --model_v1 # multi scale test -python src/test.py ctdet --houghnet --region_num 17 --vote_field_size 65 --exp_id coco_resdcn_101 --arch resdcn_101 --keep_res --resume --load_model ./models/ctdet_coco_resdcn101.pth --coco_path $COCO_PATH --flip_test --test_scales 0.6,0.8,1,1.2,1.5,1.8 +python src/test.py ctdet --houghnet --region_num 17 --vote_field_size 65 --exp_id coco_resdcn_101 --arch resdcn_101 --keep_res --resume --load_model ./models/ctdet_coco_resdcn101.pth --coco_path $COCO_PATH --flip_test --test_scales 0.6,0.8,1,1.2,1.5,1.8 --model_v1 diff --git a/experiments/ctdet_coco_resdcn101_light.sh b/experiments/ctdet_coco_resdcn101_light.sh index cbe48a0..dcbe55e 100644 --- a/experiments/ctdet_coco_resdcn101_light.sh +++ b/experiments/ctdet_coco_resdcn101_light.sh @@ -1,12 +1,14 @@ # Light Model # train -python src/main.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_resdcn_101_light --arch resdcn_101 --batch_size 44 --master_batch 8 --lr 1.75e-4 --gpus 0,1,2,3 --num_workers 16 --coco_path $COCO_PATH +python src/main.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_resdcn_101_light --arch resdcn_101 --batch_size 44 --master_batch_size 8 --lr 1.75e-4 --gpus 0,1,2,3 --num_workers 16 --coco_path $COCO_PATH + +### for testing your own trainings, please remove '--model_v1' # test -python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_resdcn_101_light --arch resdcn_101 --keep_res --resume --load_model ./models/ctdet_coco_resdcn101_light.pth --coco_path $COCO_PATH +python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_resdcn_101_light --arch resdcn_101 --keep_res --resume --load_model ./models/ctdet_coco_resdcn101_light.pth --coco_path $COCO_PATH --model_v1 # flip test -python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_resdcn_101_light --arch resdcn_101 --keep_res --resume --load_model ./models/ctdet_coco_resdcn101_light.pth --coco_path $COCO_PATH --flip_test +python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_resdcn_101_light --arch resdcn_101 --keep_res --resume --load_model ./models/ctdet_coco_resdcn101_light.pth --coco_path $COCO_PATH --flip_test --model_v1 # multi scale test -python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_resdcn_101_light --arch resdcn_101 --keep_res --resume --load_model ./models/ctdet_coco_resdcn101_light.pth --coco_path $COCO_PATH --flip_test --test_scales 0.6,0.8,1,1.2,1.5,1.8 +python src/test.py ctdet --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_resdcn_101_light --arch resdcn_101 --keep_res --resume --load_model ./models/ctdet_coco_resdcn101_light.pth --coco_path $COCO_PATH --flip_test --test_scales 0.6,0.8,1,1.2,1.5,1.8 --model_v1 diff --git a/experiments/ctseg_coco_resdcn101_baseline.sh b/experiments/ctseg_coco_resdcn101_baseline.sh new file mode 100644 index 0000000..4bbc9f3 --- /dev/null +++ b/experiments/ctseg_coco_resdcn101_baseline.sh @@ -0,0 +1,10 @@ + +### Baseline Segmentation Model + +# train +python src/main.py ctseg --dataset coco_seg --exp_id ctseg_coco_resdcn_101_baseline --arch resdcn_101 --master_batch_size 5 --batch_size 32 --lr_step 50 --num_epochs 80 --lr 1.25e-4 --gpus 0,1,2,3 --num_workers 16 --coco_path $COCO_PATH + +# test +python src/test.py ctseg --dataset coco_seg --exp_id ctseg_coco_resdcn_101_baseline --arch resdcn_101 --keep_res --resume --load_model ./models/ctseg_coco_resdcn101_baseline.pth --coco_path $COCO_PATH + + diff --git a/experiments/ctseg_coco_resdcn101_light.sh b/experiments/ctseg_coco_resdcn101_light.sh new file mode 100644 index 0000000..83ac463 --- /dev/null +++ b/experiments/ctseg_coco_resdcn101_light.sh @@ -0,0 +1,8 @@ + +### Segmentation Model with Voting + +# train +python src/main.py ctseg --dataset coco_seg --houghnet --region_num 9 --vote_field_size 17 --exp_id ctseg_coco_resdcn_101_light --arch resdcn_101 --master_batch_size 5 --batch_size 32 --lr_step 50 --num_epochs 80 --lr 1.25e-4 --gpus 0,1,2,3 --num_workers 16 --coco_path $COCO_PATH +# test +python src/test.py ctseg --dataset coco_seg --houghnet --region_num 9 --vote_field_size 17 --exp_id ctseg_coco_resdcn_101_light --arch resdcn_101 --keep_res --resume --load_model ./models/ctseg_coco_resdcn101_light.pth --coco_path $COCO_PATH + diff --git a/experiments/ddd_sub.sh b/experiments/ddd_sub.sh new file mode 100644 index 0000000..c56548a --- /dev/null +++ b/experiments/ddd_sub.sh @@ -0,0 +1,7 @@ + +# train +python src/main.py ddd --houghnet --dataset kitti --kitti_split subcnn --exp_id sub --arch dla_34 --batch_size 16 --master_batch 7 --num_epochs 70 --lr_step 45,60 --gpus 0,1 + +### for testing your own trainings, please remove '--model_v1' +# test +python src/test.py ddd --houghnet --dataset kitti --kitti_split subcnn --exp_id sub --arch dla_34 --resume --load_model ./models/ddd_kitti_dla34.pth --model_v1 diff --git a/experiments/multi_pose_hm_coco_dla34_1x.sh b/experiments/multi_pose_hm_coco_dla34_1x.sh new file mode 100644 index 0000000..5978200 --- /dev/null +++ b/experiments/multi_pose_hm_coco_dla34_1x.sh @@ -0,0 +1,11 @@ + +### voting for hm head - please be sure that voting head is 'hm' + +# train +python src/main.py multi_pose --dataset coco_hp --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_mp_voting_hm --arch dla_34 --batch_size 44 --master_batch_size 8 --lr 1.75e-4 --gpus 0,1,2,3 --num_workers 16 --load_model ./models/ctdet_coco_dla_2x.pth --coco_path $COCO_PATH + +### for testing your own trainings, please remove '--model_v1' +# test +python src/test.py multi_pose --houghnet --dataset coco_hp --exp_id coco_mp_voting_hm --arch dla_34 --keep_res --resume --load_model ./models/multi_pose_hm_coco_dla34_1x.pth --coco_path $COCO_PATH --model_v1 +# flip test +python src/test.py multi_pose --houghnet --dataset coco_hp --exp_id coco_mp_voting_hm --arch dla_34 --keep_res --resume --load_model ./models/multi_pose_hm_coco_dla34_1x.pth --coco_path $COCO_PATH --flip_test --model_v1 diff --git a/experiments/multi_pose_hm_hp_coco_dla34_1x.sh b/experiments/multi_pose_hm_hp_coco_dla34_1x.sh new file mode 100644 index 0000000..22958d7 --- /dev/null +++ b/experiments/multi_pose_hm_hp_coco_dla34_1x.sh @@ -0,0 +1,11 @@ + +### voting for both 'hm' and 'hm_hp' heads - please be sure that voting head is 'hm' and 'hm_hp' + +# train +python src/main.py multi_pose --dataset coco_hp --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_mp_voting_hm_hp --arch dla_34 --batch_size 44 --master_batch_size 8 --lr 1.75e-4 --gpus 0,1,2,3 --num_workers 16 --load_model ./models/ctdet_coco_dla_2x.pth --coco_path $COCO_PATH + +### for testing your own trainings, please remove '--model_v1' +# test +python src/test.py multi_pose --houghnet --dataset coco_hp --exp_id coco_mp_voting_hm_hp --arch dla_34 --keep_res --resume --load_model ./models/multi_pose_hm_hp_coco_dla34_1x.pth --coco_path $COCO_PATH --model_v1 +# flip test +python src/test.py multi_pose --houghnet --dataset coco_hp --exp_id coco_mp_voting_hm_hp --arch dla_34 --keep_res --resume --load_model ./models/multi_pose_hm_hp_coco_dla34_1x.pth --coco_path $COCO_PATH --flip_test --model_v1 diff --git a/experiments/multi_pose_hp_coco_dla34_1x.sh b/experiments/multi_pose_hp_coco_dla34_1x.sh new file mode 100644 index 0000000..3990955 --- /dev/null +++ b/experiments/multi_pose_hp_coco_dla34_1x.sh @@ -0,0 +1,11 @@ + +### voting for hm_hp head - please be sure that voting head is 'hm_hp' + +# train +python src/main.py multi_pose --dataset coco_hp --houghnet --region_num 9 --vote_field_size 17 --exp_id coco_mp_voting_hp --arch dla_34 --batch_size 44 --master_batch_size 8 --lr 1.75e-4 --gpus 0,1,2,3 --num_workers 16 --load_model ./models/ctdet_coco_dla_2x.pth --coco_path $COCO_PATH + +### for testing your own trainings, please remove '--model_v1' +# test +python src/test.py multi_pose --houghnet --dataset coco_hp --exp_id coco_mp_voting_hp --arch dla_34 --keep_res --resume --load_model ./models/multi_pose_hp_coco_dla34_1x.pth --coco_path $COCO_PATH --model_v1 +# flip test +python src/test.py multi_pose --houghnet --dataset coco_hp --exp_id coco_mp_voting_hp --arch dla_34 --keep_res --resume --load_model ./models/multi_pose_hp_coco_dla34_1x.pth --coco_path $COCO_PATH --flip_test --model_v1