def __read_mask(self, instance, roi):
dvid_roi = dvision.DVIDRegionOfInterest(
self.hostname,
self.port,
self.uuid,
instance)
return dvid_roi[roi.get_bounding_box()]
def __read_gt_mask(self, roi):
"""
:param roi: gunpowder.Roi
:return: uint8 np.ndarray with roi shape
"""
if self.gt_mask_roi_name is None:
raise MaskNotProvidedException
slices = roi.get_bounding_box()
dvid_roi = dvision.DVIDRegionOfInterest(self.hostname, self.port, self.uuid, self.gt_mask_roi_name)
try:
return dvid_roi[slices]
except Exception as e:
print(e)
msg = "Failure reading GT mask at slices {} with {}".format(slices, repr(self))
raise DvidSourceReadException(msg)
def __get_info(self, array_key):
if array_key in self.datasets:
data = dvision.DVIDDataInstance(self.hostname, self.port,
self.uuid,
self.datasets[array_key])
elif array_key in self.masks:
data = dvision.DVIDRegionOfInterest(self.hostname, self.port,
self.uuid,
self.masks[array_key])
else:
assert False, ("Encountered a request that is neither a volume "
"nor a mask.")
return data.info