create_masks.py 文件源码

def __init__(self, raw_image, putative_nuclei_image, putative_somata_image, centers_of_mass=None):
        super(DonutCells, self).__init__(putative_nuclei_image, raw_image, centers_of_mass)
        self.putative_somata_image = putative_somata_image
        self.putative_nuclei_image = ndimage.binary_fill_holes(self.putative_nuclei_image)
        self.watershed_image = np.logical_or(self.putative_nuclei_image, self.putative_somata_image)
        self.segmentation_labels = calculate_distance(self.centers_of_mass, self.watershed_image)
        self.labelled_nuclei = calculate_distance(self.centers_of_mass, self.putative_nuclei_image)
        self.roi_masks = create_roi_masks(self.centers_of_mass, self.putative_nuclei_image, self.putative_somata_image)