diff --git a/extensions/chaeo/products.py b/extensions/chaeo/products.py index dd9334a7e327a6c729f72f1ae18da18c8765142e..e7a12e7638c9a06a376f660c6a88b23eafd5ca67 100644 --- a/extensions/chaeo/products.py +++ b/extensions/chaeo/products.py @@ -37,6 +37,7 @@ def export_patches_from_zstack( pad_to: int = 256, make_3d: bool = False, prefix='patch', + projector=lambda x: np.max(x, axis=3, keepdims=True), **kwargs ): assert stack.chroma == 1, 'Expecting monochromatic image data' @@ -51,7 +52,7 @@ def export_patches_from_zstack( if make_3d: patch = stack.data[sl] else: - patch = np.max(stack.data[sl], axis=3, keepdims=True) + patch = projector(stack.data[sl]) assert len(patch.shape) == 4 assert patch.shape[2] == stack.chroma