diff --git a/model_server/base/accessors.py b/model_server/base/accessors.py index e05d895618a1566d4424f91f0ba5da9b14bc565f..e7af27b53791af78f3aaabd47167b5612a64c4ca 100644 --- a/model_server/base/accessors.py +++ b/model_server/base/accessors.py @@ -59,9 +59,16 @@ class GenericImageDataAccessor(ABC): ) - def get_mono(self, channel: int, mip: bool = False): + def get_mono(self, channel: int, mip: bool = False, squeeze=False): return self.get_channels([channel], mip=mip) + @property + def data_xy(self) -> np.ndarray: + if not self.nc == 1 and self.nz == 1: + raise InvalidDataShape('Can only return XY array from accessors with a single channel and single z-level') + else: + return self.data[:, :, 0, 0] + def _gc(self, channels): return self.get_channels(list(channels))