diff --git a/paddleseg/transforms/transforms.py b/paddleseg/transforms/transforms.py index 1148142a13..08167a9621 100644 --- a/paddleseg/transforms/transforms.py +++ b/paddleseg/transforms/transforms.py @@ -1207,7 +1207,7 @@ def __init__(self, min_scale_factor=0.75, max_scale_factor=1.25, im_padding_value=(128, 128, 128), - label_padding_value=(255, 255, 255)): + label_padding_value=255): self.size = size self.translation_offset = translation_offset self.max_rotation = max_rotation @@ -1264,7 +1264,8 @@ def __call__(self, im, label=None): matrix, tuple(self.size), flags=cv2.INTER_NEAREST, - borderMode=cv2.BORDER_CONSTANT) + borderMode=cv2.BORDER_CONSTANT, + borderValue=self.label_padding_value) if label is None: return (im, ) else: