Fix lensed resizing.

LCAV · Feb 29, 2024 · fc5fbf6 · fc5fbf6
1 parent ceb5d20
commit fc5fbf6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: