Fix imageio imread to V2 behaviour

ivadomed · Jul 11, 2022 · 87cf24d · 87cf24d
1 parent 55b5628
commit 87cf24d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ivadomed/loader/segmentation_pair.py b/ivadomed/loader/segmentation_pair.py
@@ -311,9 +311,9 @@ def convert_file_to_nifti(self, filename, extension, is_gt=False):
         # For '.png', '.tif', '.tiff', '.jpg' and 'jpeg' extentions
         # Read image as 8 bit grayscale in numpy array (behavior TBD in ivadomed for RGB, RBGA or higher bit depth)
         if "tif" in extension:
-            img = np.expand_dims(imageio.imread(filename, format='tiff-pil'), axis=-1).astype(np.uint8)
+            img = np.expand_dims(imageio.v2.imread(filename, format='tiff-pil'), axis=-1).astype(np.uint8)
             if len(img.shape) > 3:
-                img = np.expand_dims(imageio.imread(filename, format='tiff-pil', as_gray=True), axis=-1).astype(np.uint8)
+                img = np.expand_dims(imageio.v2.imread(filename, format='tiff-pil', as_gray=True), axis=-1).astype(np.uint8)
         else:
             img = np.expand_dims(imageio.v2.imread(filename, as_gray=True), axis=-1).astype(np.uint8)