diff --git a/ppdet/data/transform/operators.py b/ppdet/data/transform/operators.py index f2c41c12c79a4d42ef860b18cf942d2d52a0677b..b37603f87d273ede8d81ce29cd0140cbfc680572 100644 --- a/ppdet/data/transform/operators.py +++ b/ppdet/data/transform/operators.py @@ -1845,7 +1845,8 @@ class Pad(BaseOperator): assert pad_mode in [ -1, 0, 1, 2 ], 'currently only supports four modes [-1, 0, 1, 2]' - assert pad_mode == -1 and offsets, 'if pad_mode is -1, offsets should not be None' + if pad_mode == -1: + assert offsets, 'if pad_mode is -1, offsets should not be None' self.size = size self.size_divisor = size_divisor @@ -1912,7 +1913,7 @@ class Pad(BaseOperator): im_h < h and im_w < w ), '(h, w) of target size should be greater than (im_h, im_w)' else: - h = np.ceil(im_h // self.size_divisor) * self.size_divisor + h = np.ceil(im_h / self.size_divisor) * self.size_divisor w = np.ceil(im_w / self.size_divisor) * self.size_divisor if h == im_h and w == im_w: