Ensure that skimage.transform.rescale interprets image as RGB

Its `multichannel` argument is False by default now
machine-shop · Jun 25, 2020 · 2f14a61 · 2f14a61
1 parent f86f501
commit 2f14a61
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/butterfly/binarization.py b/butterfly/binarization.py
@@ -140,7 +140,8 @@ def grabcut_binarization(bfly_rgb, bfly_bin):
     bfly_region = bfly_bin_dilated_regions_sorted[0]
 
     # Downscale image to improve grabcut speed
-    bfly_rgb_rescale = rescale(bfly_rgb, GRABCUT_RESCALE_FACTOR)
+    bfly_rgb_rescale = rescale(bfly_rgb, GRABCUT_RESCALE_FACTOR,
+                               multichannel=True)
     bfly_rgb_rescale = img_as_ubyte(bfly_rgb_rescale)
 
     # Determine grabcut highlight region using butterfly region (after rescaling)