segimg.py 文件源码

def update_segvalue(self, selected_index, label):
        """
                1) mark selected_index with current label
                1) update img_arr according to seg_disp and label
                2) mark_boundaries on img_arr
                3) reset collect_points
        """
        self.seg_arr.ravel()[
            selected_index] = label  # 1) mark selected_index with current label
        tmp = self.seg_disp.reshape((-1, 3))  # 3 cols: R, G, B
        # different color represents diff labels
        tmp[selected_index, :] = self.color_map[label]
        self.seg_disp = tmp.reshape(self.seg_disp.shape)

        self.img_arr = np.array(self.ref_pic.img_arr *
                                (1 -
                                 self.alpha) +
                                self.alpha *
                                self.seg_disp, dtype=np.uint8)
        self.img_arr = np.array(
            mark_boundaries(
                self.img_arr,
                self.seg_index) * 255,
            dtype=np.uint8)
        self.collect_points = []  # reset collect_points[]
        self.update()