def _encode_label(self, labelmap): labelmap = np.array(labelmap) shape = labelmap.shape encoded_labelmap = np.ones(shape=(shape[0], shape[1]), dtype=np.float32) * 255 for i in range(len(self.configer.get('data', 'label_list'))): class_id = self.configer.get('data', 'label_list')[i] encoded_labelmap[labelmap == class_id] = i if self.configer.get('data', 'image_tool') == 'pil': encoded_labelmap = ImageHelper.to_img(encoded_labelmap.astype(np.uint8)) return encoded_labelmap
def _reduce_zero_label(self, labelmap): if not self.configer.get('data', 'reduce_zero_label'): return labelmap labelmap = np.array(labelmap) labelmap[labelmap == 0] = 255 labelmap = labelmap - 1 labelmap[labelmap == 254] = 255 if self.configer.get('data', 'image_tool') == 'pil': labelmap = ImageHelper.to_img(labelmap.astype(np.uint8)) return labelmap