testy

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