testy

1 files changed, 4 insertions, 2 deletions

diff --git a/test-mogrify.py b/test-mogrify.py
index f330230..4acabe1 100644
--- a/test-mogrify.py
+++ b/test-mogrify.py
@@ -50,13 +50,15 @@ if __name__ == '__main__':
   # output, error = process.communicate()
 
   def load_first_frame():
+    start_img_path = os.path.join(render_dir, "frame_00000.png")
     if data_opt.just_copy:
-      copyfile(opt.start_img, os.path.join(render_dir, "frame_00000.png"))
+      copyfile(opt.start_img, start_img_path)
     else:
+      print("preload {}".format(start_img))
       A_img = Image.open(opt.start_img).convert('RGB')
       A_im = np.asarray(A_img)
       A = process_image(A_im)
-      cv2.imwrite(os.path.join(render_dir, "frame_00000.png"), A)
+      cv2.imwrite(start_img_path, A)
 
     numz = re.findall(r'\d+', os.path.basename(opt.start_img))
     if len(numz) > 0: