Add pascal training commands

hiroharu-kato · Dec 1, 2019 · 6f1afa8 · 6f1afa8
1 parent 425b2dd
commit 6f1afa8
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 23 deletions.
diff --git a/README.md b/README.md
@@ -88,10 +88,11 @@ wget http://dl.caffe.berkeleyvision.org/bvlc_alexnet.caffemodel
 cd ../../
 ```
 
-Training of the provided pre-trained ShapeNet models is done by
+Training of the provided pre-trained models is done by
 
 ```shell script
 bash train_shapenet.sh
+bash train_pascal.sh
 ```
 
 ## Citation

diff --git a/train_pascal.sh b/train_pascal.sh
@@ -0,0 +1,28 @@
+export NI=80000
+export LI=1000
+
+export CID="aeroplane"
+export PARAMS="-cls ${CID} -li ${LI} -ni ${NI} -linf 0.00003 -lp 0.01 -ld 0.5 -sm 1"
+python mesh_reconstruction/train.py -ds pascal -eid pascal_proposed_color_airplane_0 $PARAMS -rs 0
+
+export CID="car"
+export PARAMS="-cls ${CID} -li ${LI} -ni ${NI} -linf 0.00003 -lp 0.01 -ld 0.5 -sm 1"
+python mesh_reconstruction/train.py -ds pascal -eid pascal_proposed_color_car $PARAMS -rs 0
+
+export CID="chair"
+export PARAMS="-cls ${CID} -li ${LI} -ni ${NI} -linf 0.00003 -lp 0.01 -ld 0.5 -sm 1"
+python mesh_reconstruction/train.py -ds pascal -eid pascal_proposed_color_chair $PARAMS -rs 0
+
+export NI=5000
+export LI=1000
+export CID="aeroplane"
+export PARAMS="-cls ${CID} -li ${LI} -ni ${NI} -linf 0.00003 -lp 0.01 -sm 1"
+python mesh_reconstruction/train.py -ds pascal -eid pascal_baseline_color_airplane $PARAMS -rs 0
+
+export CID="car"
+export PARAMS="-cls ${CID} -li ${LI} -ni ${NI} -linf 0.00003 -lp 0.01 -sm 1"
+python mesh_reconstruction/train.py -ds pascal -eid pascal_baseline_color_car $PARAMS -rs 0
+
+export CID="chair"
+export PARAMS="-cls ${CID} -li ${LI} -ni ${NI} -linf 0.00003 -lp 0.01 -sm 1"
+python mesh_reconstruction/train.py -ds pascal -eid pascal_baseline_color_chair $PARAMS -rs 0
diff --git a/train_shapenet.sh b/train_shapenet.sh
@@ -2,43 +2,43 @@
 # Single-view
 
 # silhouettes, w/o regularization
-python mesh_reconstruction/train.py -eid shapenet_single_sil_nv1 -dd $DD -ni 50000 -ds shapenet -svt 1 -nv 1 -sm 1 
+python mesh_reconstruction/train.py -eid shapenet_single_sil_nv1 -ni 50000 -ds shapenet -svt 1 -nv 1 -sm 1 
 
 # silhouettes, w/ regularization
-python mesh_reconstruction/train.py -eid shapenet_single_sil_nv1_uvr -dd $DD -ni 100000 -ds shapenet -svt 1 -nv 1 -ld 0.1 -sm 1  
-python mesh_reconstruction/train.py -eid shapenet_single_sil_nv1_uvr_cc -dd $DD -ni 100000 -ds shapenet -svt 1 -nv 1 -ld 0.2 -cc 1 -sm 1 
+python mesh_reconstruction/train.py -eid shapenet_single_sil_nv1_uvr -ni 100000 -ds shapenet -svt 1 -nv 1 -ld 0.1 -sm 1  
+python mesh_reconstruction/train.py -eid shapenet_single_sil_nv1_uvr_cc -ni 100000 -ds shapenet -svt 1 -nv 1 -ld 0.2 -cc 1 -sm 1 
 
 # color, w/o regularization
-python mesh_reconstruction/train.py -eid shapenet_single_color_nv1 -dd $DD -ni 50000 -ds shapenet -svt 1 -nv 1 -lp 0.5 -sm 1 
+python mesh_reconstruction/train.py -eid shapenet_single_color_nv1 -ni 50000 -ds shapenet -svt 1 -nv 1 -lp 0.5 -sm 1 
 
 # color, w/ regularization
-python mesh_reconstruction/train.py -eid shapenet_single_color_nv1_uvr -dd $DD -ni 100000 -ds shapenet -svt 1 -nv 1 -lp 0.5 -ld 2 -sm 1 
-python mesh_reconstruction/train.py -eid shapenet_single_color_nv1_uvr_cc -dd $DD -ni 100000 -ds shapenet -svt 1 -nv 1 -lp 0.5 -ld 2 -cc 1 -sm 1  
+python mesh_reconstruction/train.py -eid shapenet_single_color_nv1_uvr -ni 100000 -ds shapenet -svt 1 -nv 1 -lp 0.5 -ld 2 -sm 1 
+python mesh_reconstruction/train.py -eid shapenet_single_color_nv1_uvr_cc -ni 100000 -ds shapenet -svt 1 -nv 1 -lp 0.5 -ld 2 -cc 1 -sm 1  
 
 ####################################################################################################
 # Multi-view
 
 # silhouettes
-python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv20 -dd $DD -ni 500000 -ds shapenet -sm 1  
-python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv10 -dd $DD -ni 250000 -ds shapenet -nv 10 -sm 1 
-python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv5 -dd $DD -ni 125000 -ds shapenet -nv 5 -sm 1 
-python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv3 -dd $DD -ni 75000 -ds shapenet -nv 3 -sm 1 
-python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv2 -dd $DD -ni 50000 -ds shapenet -nv 2 -sm 1 
+python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv20 -ni 500000 -ds shapenet -sm 1  
+python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv10 -ni 250000 -ds shapenet -nv 10 -sm 1 
+python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv5 -ni 125000 -ds shapenet -nv 5 -sm 1 
+python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv3 -ni 75000 -ds shapenet -nv 3 -sm 1 
+python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv2 -ni 50000 -ds shapenet -nv 2 -sm 1 
 
 # silhouettes + regularization
-python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv20_uvr_cc_long -dd $DD -ni 1000000 -ds shapenet -ld 0.03 -cc 1 -sm 1
-python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv20_uvr_cc -dd $DD -ni 500000 -ds shapenet -ld 0.03 -cc 1 -sm 1
-python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv10_uvr_cc -dd $DD -ni 500000 -ds shapenet -nv 10 -ld 0.03 -cc 1 -sm 1
-python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv5_uvr_cc -dd $DD -ni 250000 -ds shapenet -nv 5 -ld 0.03 -cc 1 -sm 1
-python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv3_uvr_cc -dd $DD -ni 150000 -ds shapenet -nv 3 -ld 0.03 -cc 1 -sm 1
-python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv2_uvr_cc -dd $DD -ni 100000 -ds shapenet -nv 2 -ld 0.03 -cc 1 -sm 1
+python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv20_uvr_cc_long -ni 1000000 -ds shapenet -ld 0.03 -cc 1 -sm 1
+python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv20_uvr_cc -ni 500000 -ds shapenet -ld 0.03 -cc 1 -sm 1
+python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv10_uvr_cc -ni 500000 -ds shapenet -nv 10 -ld 0.03 -cc 1 -sm 1
+python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv5_uvr_cc -ni 250000 -ds shapenet -nv 5 -ld 0.03 -cc 1 -sm 1
+python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv3_uvr_cc -ni 150000 -ds shapenet -nv 3 -ld 0.03 -cc 1 -sm 1
+python mesh_reconstruction/train.py -eid shapenet_multi_sil_nv2_uvr_cc -ni 100000 -ds shapenet -nv 2 -ld 0.03 -cc 1 -sm 1
 
 # color
-python mesh_reconstruction/train.py -eid shapenet_multi_color_nv20 -dd $DD -ni 500000 -ds shapenet -lp 0.1 -sm 1 
-python mesh_reconstruction/train.py -eid shapenet_multi_color_nv2 -dd $DD -ni 50000 -ds shapenet -lp 0.1 -nv 2 -sm 1  
+python mesh_reconstruction/train.py -eid shapenet_multi_color_nv20 -ni 500000 -ds shapenet -lp 0.1 -sm 1 
+python mesh_reconstruction/train.py -eid shapenet_multi_color_nv2 -ni 50000 -ds shapenet -lp 0.1 -nv 2 -sm 1  
 
 # color + regularization
-python mesh_reconstruction/train.py -eid shapenet_multi_color_nv20_uvr_cc_long -dd $DD -ni 1000000 -ds shapenet -lp 0.1 -nv 20 -ld 0.3 -cc 1 -sm 1
-python mesh_reconstruction/train.py -eid shapenet_multi_color_nv20_uvr_cc -dd $DD -ni 500000 -ds shapenet -lp 0.1 -nv 20 -ld 0.3 -cc 1 -sm 1
-python mesh_reconstruction/train.py -eid shapenet_multi_color_nv2_uvr_cc -dd $DD -ni 100000 -ds shapenet -lp 0.1 -nv 2 -ld 0.3 -cc 1 -sm 1
+python mesh_reconstruction/train.py -eid shapenet_multi_color_nv20_uvr_cc_long -ni 1000000 -ds shapenet -lp 0.1 -nv 20 -ld 0.3 -cc 1 -sm 1
+python mesh_reconstruction/train.py -eid shapenet_multi_color_nv20_uvr_cc -ni 500000 -ds shapenet -lp 0.1 -nv 20 -ld 0.3 -cc 1 -sm 1
+python mesh_reconstruction/train.py -eid shapenet_multi_color_nv2_uvr_cc -ni 100000 -ds shapenet -lp 0.1 -nv 2 -ld 0.3 -cc 1 -sm 1