diff options
Diffstat (limited to 'test-mogrify.py')
| -rw-r--r-- | test-mogrify.py | 6 |
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: |