diff --git a/ppdet/data/transform/operators.py b/ppdet/data/transform/operators.py index b37603f87d273ede8d81ce29cd0140cbfc680572..54f918c9bd46f8283c3be9ef57e4589f09481b51 100644 --- 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) diff --git a/static/ppdet/data/transform/operators.py b/static/ppdet/data/transform/operators.py index 78ae281e9c1c652a319eda9bf354ee214934773b..d51e68d179e39ef1f372bb246025c4eaf5e235fd 100644 --- 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)