diff --git a/model_server/accessors.py b/model_server/accessors.py index d9211c4876c21e24ac4330af4de5806193a215a5..7183036a63c653328315761e89c066a30fc64d64 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): """