disable matplotlib and make parser correct

hlzz · hlzz · commit 5abb4bc34811 · 2018-10-13T02:00:21.000+08:00
diff --git a/mvsnet/test.py b/mvsnet/test.py
@@ -14,7 +14,7 @@
 import numpy as np
 
 import cv2
-import matplotlib.pyplot as plt
+#import matplotlib.pyplot as plt
 import tensorflow as tf
 
 sys.path.append("../")
diff --git a/mvsnet/train.py b/mvsnet/train.py
@@ -26,9 +26,6 @@
 from model import inference, mvsnet_loss, depth_refine
 from homography_warping import get_homographies, homography_warping
 
-FLAGS = tf.app.flags.FLAGS
-
-
 # params for datasets
 tf.app.flags.DEFINE_string('dtu_data_root', '/data/dtu/', 
                            """Path to dtu dataset.""")
@@ -86,6 +83,8 @@
 tf.app.flags.DEFINE_float('gamma', 0.9,
                           """Learning rate decay rate.""")
 
+FLAGS = tf.app.flags.FLAGS
+
 class MVSGenerator:
     """ data generator class, tf only accept generator without param """
     def __init__(self, sample_list, view_num):
@@ -306,15 +305,5 @@ def main(argv=None):  # pylint: disable=unused-argument
 
 
 if __name__ == '__main__':
-
-    parser = argparse.ArgumentParser()
-    parser.add_argument('--ckpt_step', type=int, default = FLAGS.ckpt_step)
-    parser.add_argument('--view_num', type=int, default = FLAGS.view_num)
-    args = parser.parse_args()
-
-    FLAGS.ckpt_step = args.ckpt_step
-    FLAGS.view_num = args.view_num
-    print ('Training MVSNet with %d views' % args.view_num)
-
-
+    print ('Training MVSNet with %d views' % FLAGS.view_num)
     tf.app.run()
