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
