fix np.int warning in operator (#3344)

9d942c50 · Guanghua Yu · GitHub · 18596b01 · 9d942c50 · 9d942c50
隐藏空白更改
内联并排

Showing with 4 addition and 4 deletion

ppdet/data/transform/operators.py ppdet/data/transform/operators.py +2 -2

static/ppdet/data/transform/operators.py static/ppdet/data/transform/operators.py +2 -2

未找到文件。
--- a/ppdet/data/transform/operators.py
+++ b/ppdet/data/transform/operators.py
@@ -1506,8 +1506,8 @@ class Cutmix(BaseOperator):
        w = max(img1.shape[1], img2.shape[1])
        cut_rat = np.sqrt(1. - factor)

-        cut_w = np.int(w * cut_rat)
-        cut_h = np.int(h * cut_rat)
+        cut_w = np.int32(w * cut_rat)
+        cut_h = np.int32(h * cut_rat)

        # uniform
        cx = np.random.randint(w)

--- a/static/ppdet/data/transform/operators.py
+++ b/static/ppdet/data/transform/operators.py
@@ -1343,8 +1343,8 @@ class CutmixImage(BaseOperator):
            w = max(img1.shape[1], img2.shape[1])
            cut_rat = np.sqrt(1. - factor)

-            cut_w = np.int(w * cut_rat)
-            cut_h = np.int(h * cut_rat)
+            cut_w = np.int32(w * cut_rat)
+            cut_h = np.int32(h * cut_rat)

            # uniform
            cx = np.random.randint(w)