testy

2 files changed, 4 insertions, 1 deletions

diff --git a/options/test_options.py b/options/test_options.py
index 418d5e6..6f38eef 100644
--- a/options/test_options.py
+++ b/options/test_options.py
@@ -12,5 +12,5 @@ class TestOptions(BaseOptions):
         self.parser.add_argument('--phase', type=str, default='test', help='train, val, test, etc')
         self.parser.add_argument('--which_epoch', type=str, default='latest', help='which epoch to load? set to latest to use latest cached model')
         self.parser.add_argument('--how_many', type=int, default=50, help='how many test images to run')
-        self.parser.add_argument('--mov', type=str, help='filename of movie to export')
+        self.parser.add_argument('--mov_file', type=str, help='filename of movie to export')
         self.isTrain = False
diff --git a/test-mogrify.py b/test-mogrify.py
index c1d6c2f..0efabe1 100644
--- a/test-mogrify.py
+++ b/test-mogrify.py
@@ -111,6 +111,9 @@ if __name__ == '__main__':
       frame_fn = "ren_%05d.png"
       video_fn = tag + "_mogrify.mp4"
 
+    if opt.mov_file:
+      video_fn = opt.mov_file + '.mp4'
+
     cmd = ("ffmpeg", "-i", render_dir + frame_fn, "-y", "-c:v", "libx264", "-vf", "fps=30", "-pix_fmt", "yuv420p", "-s", "456x256", render_dir + video_fn)
     process = subprocess.Popen(cmd, stdout=subprocess.PIPE)
     output, error = process.communicate()