Fix resizing.

LCAV · Oct 24, 2023 · 2d2a10c · 2d2a10c
1 parent 6065ee8
commit 2d2a10c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/lensless/utils/image.py b/lensless/utils/image.py
@@ -66,7 +66,7 @@ def resize(img, factor=None, shape=None, interpolation=cv2.INTER_CUBIC):
     else:
         resized = np.array(
             [
-                cv2.resize(img[i], dsize=new_shape[::-1], interpolation=interpolation)
+                cv2.resize(img[i], dsize=tuple(new_shape[::-1]), interpolation=interpolation)
                 for i in range(img.shape[-4])
             ]
         )

diff --git a/scripts/measure/collect_dataset_on_device.py b/scripts/measure/collect_dataset_on_device.py
@@ -214,9 +214,9 @@ def collect_dataset(config):
                     )
 
                     if down:
-                        output = resize(output[None, ...], 1 / down, interpolation=cv2.INTER_CUBIC)[
-                            0
-                        ]
+                        output = resize(
+                            output[None, ...], factor=1 / down, interpolation=cv2.INTER_CUBIC
+                        )[0]
 
                     # print range
                     print(f"{output_fp}, range: {output.min()} - {output.max()}")