Merge pull request MedMNIST#29 from orientnab/orientnab-patch-1-monta…

…ge2d Update montage2d to scikit-image 0.20
yqchupy · Apr 2, 2023 · 50cc28c · 50cc28c
2 parents 16e3ead + 457c2d2
commit 50cc28c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/medmnist/utils.py b/medmnist/utils.py
@@ -20,9 +20,9 @@ def save2d(imgs, labels, img_folder,
 
 
 def montage2d(imgs, n_channels, sel):
-
     sel_img = imgs[sel]
-    montage_arr = skimage_montage(sel_img, multichannel=(n_channels == 3))
+    channel_axis = 3 if n_channels == 3 else None                                                                                          
+    montage_arr = skimage_montage(sel_img, channel_axis=channel_axis)
     montage_img = Image.fromarray(montage_arr)
 
     return montage_img