Skip to content

Commit 1baf96e

Browse files
authored
[Fix] Fix the problem of label_padding_value not work in warpAffine (#2072)
1 parent 75a6cf5 commit 1baf96e

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

paddleseg/transforms/transforms.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -1207,7 +1207,7 @@ def __init__(self,
12071207
min_scale_factor=0.75,
12081208
max_scale_factor=1.25,
12091209
im_padding_value=(128, 128, 128),
1210-
label_padding_value=(255, 255, 255)):
1210+
label_padding_value=255):
12111211
self.size = size
12121212
self.translation_offset = translation_offset
12131213
self.max_rotation = max_rotation
@@ -1264,7 +1264,8 @@ def __call__(self, im, label=None):
12641264
matrix,
12651265
tuple(self.size),
12661266
flags=cv2.INTER_NEAREST,
1267-
borderMode=cv2.BORDER_CONSTANT)
1267+
borderMode=cv2.BORDER_CONSTANT,
1268+
borderValue=self.label_padding_value)
12681269
if label is None:
12691270
return (im, )
12701271
else:

0 commit comments

Comments
 (0)