def _reduce_zero_label(self, labelmap):
if not self.configer.get('data', 'reduce_zero_label'):
return labelmap
labelmap = np.array(labelmap)
encoded_labelmap = labelmap - 1
if self.configer.get('data', 'image_tool') == 'pil':
encoded_labelmap = ImageHelper.np2img(
encoded_labelmap.astype(np.uint8))
return encoded_labelmap
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.np2img(
encoded_labelmap.astype(np.uint8))
return encoded_labelmap