瀏覽代碼

use normal division in padding transfrom

will-jl944 4 年之前
父節點
當前提交
5911e3aff8
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      dygraph/paddlex/cv/transforms/operators.py

+ 1 - 1
dygraph/paddlex/cv/transforms/operators.py

@@ -1004,7 +1004,7 @@ class Padding(Transform):
             ), 'target size ({}, {}) cannot be less than image size ({}, {})'\
             ), 'target size ({}, {}) cannot be less than image size ({}, {})'\
                 .format(h, w, im_h, im_w)
                 .format(h, w, im_h, im_w)
         else:
         else:
-            h = (np.ceil(im_h // self.size_divisor) *
+            h = (np.ceil(im_h / self.size_divisor) *
                  self.size_divisor).astype(int)
                  self.size_divisor).astype(int)
             w = (np.ceil(im_w / self.size_divisor) *
             w = (np.ceil(im_w / self.size_divisor) *
                  self.size_divisor).astype(int)
                  self.size_divisor).astype(int)