Expanded 3d patch stack as 3d multichannel patch stack

extensions/chaeo/products.py

@@ -9,7 +9,7 @@ from skimage.io import imsave
 from skimage.measure import find_contours, shannon_entropy
 from tifffile import imwrite
 
-from extensions.chaeo.accessors import MonoPatchStack, PatchStack3D
+from extensions.chaeo.accessors import MonoPatchStack, Multichannel3dPatchStack
 from extensions.chaeo.annotators import draw_box_on_patch, draw_contours_on_patch
 from model_server.accessors import GenericImageDataAccessor, InMemoryDataAccessor
 from model_server.process import pad, rescale, resample_to_8bit
@@ -197,10 +197,10 @@ def get_patches_from_zmask_meta(
             patch = pad(patch, pad_to)
 
         patches.append(patch)
-    if not make_3d:
+    if not make_3d and pc == 1:
         return MonoPatchStack(patches)
     else:
-        return PatchStack3D(patches)
+        return Multichannel3dPatchStack(patches)
 
 def export_patches_from_zstack(
         where: Path,