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