From 060f6991aa1e072bfa3f7128cbd3c426253465af Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Tue, 9 Apr 2024 15:50:55 +0200
Subject: [PATCH] Allow multichannel raw and single-channel mask to pass
assertion
---
model_server/base/models.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/model_server/base/models.py b/model_server/base/models.py
index 4fa08490..a8576d40 100644
--- a/model_server/base/models.py
+++ b/model_server/base/models.py
@@ -85,7 +85,7 @@ class InstanceSegmentationModel(ImageToImageModel):
"""
if not mask.is_mask():
raise InvalidInputImageError('Expecting a binary mask')
- if not img.shape == mask.shape:
+ if img.hw != mask.hw or img.nz != mask.nz:
raise InvalidInputImageError('Expect input image and mask to be the same shape')
def label_patch_stack(self, img: PatchStack, mask: PatchStack, **kwargs):
--
GitLab