From 5919b052eed87b1ef348f000a0c48c93aab95f39 Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Fri, 29 Sep 2023 09:08:32 +0200
Subject: [PATCH] Shorthand getters for data dimensions
---
model_server/accessors.py | 16 ++++++++++++++++
1 file changed, 16 insertions(+)
diff --git a/model_server/accessors.py b/model_server/accessors.py
index d9211c48..7183036a 100644
--- a/model_server/accessors.py
+++ b/model_server/accessors.py
@@ -39,6 +39,22 @@ class GenericImageDataAccessor(ABC):
c = int(channel)
return InMemoryDataAccessor(self.data[:, :, c:(c+1), :])
+ @property
+ def dtype(self):
+ return self.data.dtype
+
+ @property
+ def hw(self):
+ """
+ Get data height and width as a tuple
+ :return: tuple of (Y, X) dimensions
+ """
+ return self.shape_dict['Y'], self.shape_dict['X']
+
+ @property
+ def nz(self):
+ return self.shape_dict['Z']
+
@property
def data(self):
"""
--
GitLab