add demo code

shurans · Dec 7, 2016 · 6e1fd82 · 6e1fd82
1 parent be3ed50
commit 6e1fd82
Show file tree

Hide file tree

Showing 11 changed files with 8,032 additions and 13 deletions.
diff --git a/.gitignore b/.gitignore
@@ -2,6 +2,8 @@
 *.caffemodel
 
 *.solverstate
+*.hdf5
+
 
 .smb*
 

diff --git a/README.md b/README.md
@@ -70,11 +70,18 @@ The code organized as follow
     export PYTHONPATH=~/build_master_release/python:$PYTHONPATH
     ```
 
+### Quick Demo:
+  ```Shell 
+  cd demo
+  python demotest_model.py
+  ```
+This demo runs semantic scene compeletion on one NYU depth map using pertrained model and outputs one 'ply' visulization of the result
+
 
 ### Testing:
 0. run the testing script
-    cd caffe_code/script/test
-    python test_model.py
+  ` cd caffe_code/script/test
+    python test_model.py`
 0. the output result will be stored in results in .hdf5 format
 0. To test on other testset, e.g. suncg, nyu, nyucad you need to modify paths in “test_model.py”.
 

diff --git a/caffe_code/script/test/deploy_net_nyu.txt b/caffe_code/script/test/deploy_net_nyu.txt
@@ -15,9 +15,9 @@ layer {
     crop_size: 240
     crop_size: 144
     crop_size: 240
-    label_size: 240
-    label_size: 144
-    label_size: 240
+    label_size: 60
+    label_size: 36
+    label_size: 60
     seg_classes: 11
     shuffle: 0
     occ_empty_only: true

diff --git a/caffe_code/script/test/deploy_net_nyucad.txt b/caffe_code/script/test/deploy_net_nyucad.txt
@@ -15,9 +15,9 @@ layer {
     crop_size: 240
     crop_size: 144
     crop_size: 240
-    label_size: 240
-    label_size: 144
-    label_size: 240
+    label_size: 60
+    label_size: 36
+    label_size: 60
     seg_classes: 11
     shuffle: 0
     occ_empty_only: true

diff --git a/caffe_code/script/test/deploy_net_suncg.txt b/caffe_code/script/test/deploy_net_suncg.txt
@@ -15,9 +15,9 @@ layer {
     crop_size: 240
     crop_size: 144
     crop_size: 240
-    label_size: 240
-    label_size: 144
-    label_size: 240
+    label_size: 60
+    label_size: 36
+    label_size: 60
     seg_classes: 11
     shuffle: false
     occ_empty_only: true

diff --git a/caffe_code/script/train/trainsuncg/train_net.txt b/caffe_code/script/train/trainsuncg/train_net.txt
@@ -5,8 +5,12 @@ layer {
   top: "seg_label"
   top: "seg_weight"
   suncg_data_param {
-    file_data: "../../../../data/depthbin/SUNCG01"
-    file_data: "../../../../data/depthbin/SUNCG02"
+    file_data: "../../../../data/depthbin/SUNCGtrain_1_500"
+    file_data: "../../../../data/depthbin/SUNCGtrain_501_1000"
+    file_data: "../../../../data/depthbin/SUNCGtrain_1001_2000"
+    file_data: "../../../../data/depthbin/SUNCGtrain_1001_3000"
+    file_data: "../../../../data/depthbin/SUNCGtrain_3001_5000"
+    file_data: "../../../../data/depthbin/SUNCGtrain_5001_7000"
     file_list: ""
     vox_unit: 0.02
     vox_margin: 0.24

diff --git a/demo/NYU0001_0000.bin b/demo/NYU0001_0000.bin
diff --git a/demo/NYU0001_0000.png b/demo/NYU0001_0000.png