From fc5fbf65244cd53fcef720af791e2fd994e73d83 Mon Sep 17 00:00:00 2001
From: Eric Bezzam <ebezzam@gmail.com>
Date: Thu, 29 Feb 2024 15:20:43 +0100
Subject: [PATCH] Fix lensed resizing.

---
 lensless/utils/dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lensless/utils/dataset.py b/lensless/utils/dataset.py
index 3ea548dd..14dd2629 100644
--- a/lensless/utils/dataset.py
+++ b/lensless/utils/dataset.py
@@ -1050,7 +1050,7 @@ def _get_images_pair(self, idx):
         if self.alignment is not None:
             lensed_np = resize(
                 lensed_np,
-                shape=(self.alignment["height"], self.alignment["width"]),
+                shape=(self.alignment["height"], self.alignment["width"], 3),
                 interpolation=cv2.INTER_NEAREST,
             )
         elif self.display_res is not None: