Skip to content

Commit 18596b0

Browse files
authored
fix bug in Pad transform (PaddlePaddle#2569)
1 parent 796874c commit 18596b0

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

ppdet/data/transform/operators.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1845,7 +1845,8 @@ def __init__(self,
18451845
assert pad_mode in [
18461846
-1, 0, 1, 2
18471847
], 'currently only supports four modes [-1, 0, 1, 2]'
1848-
assert pad_mode == -1 and offsets, 'if pad_mode is -1, offsets should not be None'
1848+
if pad_mode == -1:
1849+
assert offsets, 'if pad_mode is -1, offsets should not be None'
18491850

18501851
self.size = size
18511852
self.size_divisor = size_divisor
@@ -1912,7 +1913,7 @@ def apply(self, sample, context=None):
19121913
im_h < h and im_w < w
19131914
), '(h, w) of target size should be greater than (im_h, im_w)'
19141915
else:
1915-
h = np.ceil(im_h // self.size_divisor) * self.size_divisor
1916+
h = np.ceil(im_h / self.size_divisor) * self.size_divisor
19161917
w = np.ceil(im_w / self.size_divisor) * self.size_divisor
19171918

19181919
if h == im_h and w == im_w:

0 commit comments

Comments
 (0)