fix bug in RandCropImageV2

2022-06-20 10:53:19 +08:00 · 2022-06-20 10:53:19 +08:00 · 3b1247d069
parent 851e915e6a
commit 3b1247d069
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