fix parameters

YaoYao · YaoYao · commit 8680ac372f88 · 2019-07-27T13:02:45.000+08:00
diff --git a/README.md b/README.md
@@ -33,9 +33,9 @@
 * Enter the ``MVSNet/mvsnet`` folder, in ``train.py``, set ``dtu_data_root`` to your ``MVS_TRANING`` path
 * Create a log folder and a model folder in wherever you like to save the training outputs. Set the ``log_dir`` and ``save_dir`` in ``train.py`` correspondingly
 * Train MVSNet (GTX1080Ti): 
-``python train.py --regularization 'GRU'`` 
+``python train.py --regularization '3DCNNs'`` 
 * Train R-MVSNet (GTX1080Ti):
-``python train.py --regularization '3DCNNs'``
+``python train.py --regularization 'GRU'``
 
 ### Testing
 
diff --git a/mvsnet/depthfusion.py b/mvsnet/depthfusion.py
@@ -92,7 +92,7 @@ def mvsnet_to_gipuma_cam(in_path, out_path):
 
     return
 
-def fake_colmap_normal(in_depth_path, out_normal_path):
+def fake_gipuma_normal(in_depth_path, out_normal_path):
     
     depth_image = read_gipuma_dmb(in_depth_path)
     image_shape = np.shape(depth_image)
@@ -155,8 +155,7 @@ def mvsnet_to_gipuma(dense_folder, gipuma_point_folder):
         out_depth_dmb = os.path.join(sub_depth_folder, 'disp.dmb')
         fake_normal_dmb = os.path.join(sub_depth_folder, 'normals.dmb')
         mvsnet_to_gipuma_dmb(in_depth_pfm, out_depth_dmb)
-        fake_colmap_normal(out_depth_dmb, fake_normal_dmb)
-
+        fake_gipuma_normal(out_depth_dmb, fake_normal_dmb)
 
 def probability_filter(dense_folder, prob_threshold):
     image_folder = os.path.join(dense_folder, 'images')
@@ -176,7 +175,6 @@ def probability_filter(dense_folder, prob_threshold):
         depth_map[prob_map < prob_threshold] = 0
         write_pfm(out_depth_map_path, depth_map)
 
-
 def depth_map_fusion(point_folder, fusibile_exe_path, disp_thresh, num_consistent):
 
     cam_folder = os.path.join(point_folder, 'cams')
