diff options
| -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 19bf0cc..570cdb9 100644 --- a/test-mogrify.py +++ b/test-mogrify.py @@ -53,7 +53,8 @@ if __name__ == '__main__': if data_opt.just_copy: copyfile(opt.start_img, render_dir + "frame_00000.png") else: - A_im = Image.open(opt.start_img).convert('RGB') + A_img = Image.open(opt.start_img).convert('RGB') + A_im = numpy.asarray(A_img) A = process_image(A_im) cv2.imwrite(render_dir + "frame_00000.png", img) @@ -123,7 +124,8 @@ if __name__ == '__main__': tmp_im = im.copy() if data_opt.sequence: - A_im = Image.open(sequence_path).convert('RGB') + A_img = Image.open(sequence_path).convert('RGB') + A_im = numpy.asarray(A_img) frac_a = data_opt.recursive_frac frac_b = data_opt.sequence_frac frac_c = 1.0 - frac_a - frac_b |