1 files changed, 2 insertions, 1 deletions
diff --git a/test.py b/test.py
index 2e8863b..858db98 100644
--- a/test.py
+++ b/test.py
@@ -61,7 +61,8 @@ if __name__ == '__main__':
 
     os.remove(render_dir + "frame_00000.png")
 
-    cmd = ("/usr/bin/ffmpeg", "-i", render_dir + "frame_%05d.png", "-c:v", "libx264", "-vf", "fps=30", "-pix_fmt", "yuv420p", render_dir + opt.name + "_" + opt.experiment + ".mp4")
+
+    cmd = ("/usr/bin/ffmpeg", "-i", render_dir + "test_latest/images/frame_%05d_fake_B.png", "-c:v", "libx264", "-vf", "fps=30", "-pix_fmt", "yuv420p", render_dir + opt.name + "_" + opt.experiment + ".mp4")
     process = subprocess.Popen(cmd, stdout=subprocess.PIPE)
     output, error = process.communicate()