Merge pull request #2086 from HydrogenSulfate/fix_RandCropImageV2

fix bug in RandCropImageV2
2022-06-20 19:49:49 +08:00 · 2022-06-20 19:49:49 +08:00 · 323c0cb1d5
parent 851e915e6a 3b1247d069
commit 323c0cb1d5
1 changed files with 1 additions and 1 deletions
--- a/ppcls/data/preprocess/ops/operators.py
+++ b/ppcls/data/preprocess/ops/operators.py
@ -430,7 +430,7 @@ class RandCropImageV2(object):

    def __call__(self, img):
        if isinstance(img, np.ndarray):
-            img_h, img_w = img.shap[0], img.shap[1]
+            img_h, img_w = img.shape[0], img.shape[1]
        else:
            img_w, img_h = img.size
        tw, th = self.size