change option no_visualize to do_visualize

Author: walkoncross

options/test_options.py

@@ -1,5 +1,4 @@
-"""This script contains the test options for Deep3DFaceRecon_pytorch
-"""
+"""This script contains the test options for Deep3DFaceRecon_pytorch"""
 
 from .base_options import BaseOptions
 
@@ -12,10 +11,40 @@ class TestOptions(BaseOptions):
 
     def initialize(self, parser):
         parser = BaseOptions.initialize(self, parser)  # define shared options
-        parser.add_argument('--phase', type=str, default='test', help='train, val, test, etc')
-        parser.add_argument('--dataset_mode', type=str, default=None, help='chooses how datasets are loaded. [None | flist]')
-        parser.add_argument('--img_folder', type=str, default='examples', help='folder for test images.')
-        parser.add_argument('--no_visualize', '--no_vis', '--no_viz', action='store_true', help='do not render/visualize the results.')
+        parser.add_argument(
+            "--phase",
+            type=str,
+            default="test",
+            help="train, val, test, etc",
+        )
+        parser.add_argument(
+            "--dataset_mode",
+            type=str,
+            default=None,
+            help="chooses how datasets are loaded. [None | flist]",
+        )
+        parser.add_argument(
+            "--img_folder", type=str, default="examples", help="folder for test images."
+        )
+        
+        group = parser.add_mutually_exclusive_group(required=False)
+        group.add_argument(
+            "--visualize",
+            "--vis",
+            "--viz",
+            dest="do_visualize",
+            action="store_true",
+            default=True,
+            help="render/visualize the results.",
+        )
+        group.add_argument(
+            "--no_visualize",
+            "--no_vis",
+            "--no_viz",
+            dest="do_visualize",
+            action="store_false",
+            help="do not render/visualize the results.",
+        )
 
         # Dropout and Batchnorm has different behavior during training and test.
         self.isTrain = False

test.py

@@ -69,7 +69,7 @@ def main(opt):
     model.parallelize()
     model.eval()
 
-    if not opt.no_visualize:
+    if opt.do_visualize:
         assert (
             model.renderer is not None
         ), "Visualization is only supported for models with a renderer"
@@ -103,9 +103,9 @@ def main(opt):
             "lms": lm_tensor,
         }
         model.set_input(data)  # unpack data from data loader
-        model.test(do_render=(not opt.no_visualize))  # run inference
+        model.test(do_render=opt.do_visualize)  # run inference
 
-        if not opt.no_visualize:
+        if opt.do_visualize:
             visuals = model.get_current_visuals()  # get image results
             visualizer.display_current_results(
                 visuals,