From 2bc91576cd337bc2a1c8d43947ba60bf9414095b Mon Sep 17 00:00:00 2001 From: songyi Date: Tue, 10 Dec 2019 18:28:56 +0800 Subject: [PATCH] Update deblur_image.py --- scripts/deblur_image.py | 1 + 1 file changed, 1 insertion(+) diff --git a/scripts/deblur_image.py b/scripts/deblur_image.py index 0ad4761..56d4fdc 100644 --- a/scripts/deblur_image.py +++ b/scripts/deblur_image.py @@ -13,6 +13,7 @@ def deblur(weight_path, input_dir, output_dir): g.load_weights(weight_path) for image_name in os.listdir(input_dir): image = np.array([preprocess_image(load_image(os.path.join(input_dir, image_name)))]) + image=image[:,:,:, [0, 1, 2]] x_test = image generated_images = g.predict(x=x_test) generated = np.array([deprocess_image(img) for img in generated_images])