train and test scripts for instance segmentation, keypoint detection …

…and 3D object detection are added. others are updated as well.

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
 
 
 

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
 
 
 

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
 
 
 

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

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
 

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
 
 

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

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
+

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

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

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

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