From d29a9d6b99ba48c4cbead4da9a50274f3247e343 Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Thu, 1 Aug 2024 12:08:36 +0200
Subject: [PATCH] Added convenience property to get XYZ ndarray on mono arrays
---
model_server/base/accessors.py | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/model_server/base/accessors.py b/model_server/base/accessors.py
index e7af27b5..cd35da8a 100644
--- a/model_server/base/accessors.py
+++ b/model_server/base/accessors.py
@@ -69,6 +69,13 @@ class GenericImageDataAccessor(ABC):
else:
return self.data[:, :, 0, 0]
+ @property
+ def data_xyz(self) -> np.ndarray:
+ if not self.nc == 1:
+ raise InvalidDataShape('Can only return XYZ array from accessors with a single channel')
+ else:
+ return self.data[:, :, 0, :]
+
def _gc(self, channels):
return self.get_channels(list(channels))
--
GitLab