megafacade.py 文件源码

def load_image(self, path):
        self.data = imread(path)
        self.path = path
        self._load_image_mask()
        self._mask_out_common_obstructions()
        self._rectify_image()

        self.driving_layers = driving.process_strip(channels_first(self.rectified * 255))
        self.facade_layers = i12.process_strip(channels_first(self.rectified * 255))

        self._create_sky_mask()
        self._segment_windows()
        self._segment_facade_edges()

        facade_cuts = self._split_at_facade_edges()
        facade_mask = self._create_facade_mask()
        wall_colors = self._mask_out_wall_colors(facade_mask)
        self.wall_colors = wall_colors

        self.facade_candidates = self._find_facade_candidates(wall_colors, facade_cuts)