diff options
| -rw-r--r-- | test-mogrify.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/test-mogrify.py b/test-mogrify.py index d33b36c..4596f2c 100644 --- a/test-mogrify.py +++ b/test-mogrify.py @@ -56,7 +56,7 @@ if __name__ == '__main__': A_img = Image.open(opt.start_img).convert('RGB') A_im = np.asarray(A_img) A = process_image(A_im) - cv2.imwrite(render_dir + "frame_00000.png", img) + cv2.imwrite(render_dir + "frame_00000.png", A) numz = re.findall(r'\d+', os.path.basename(opt.start_img)) if len(numz) > 0: @@ -115,6 +115,8 @@ if __name__ == '__main__': render_path = render_dir + "ren_{:05d}.png".format(i+1) sequence_path = A_dir.format(A_offset+i+1) # A_offset + + # save rendered image image_pil = Image.fromarray(im, mode='RGB') image_pil.save(tmp_path) os.rename(tmp_path, render_path) @@ -151,8 +153,8 @@ if __name__ == '__main__': tmp_im = im.copy().astype('uint8') #print(im.shape, im.dtype) - image_pil = Image.fromarray(im, mode='RGB') - im = np.asarray(image_pil).astype('uint8') + # image_pil = Image.fromarray(im, mode='RGB') + # im = np.asarray(image_pil).astype('uint8') #print(im.shape, im.dtype) img = process_image(im) |