mirror of
https://github.com/PaddlePaddle/PaddleClas.git
synced 2025-06-03 21:55:06 +08:00
fix bug in RandCropImageV2
This commit is contained in:
parent
851e915e6a
commit
3b1247d069
@ -430,7 +430,7 @@ class RandCropImageV2(object):
|
|||||||
|
|
||||||
def __call__(self, img):
|
def __call__(self, img):
|
||||||
if isinstance(img, np.ndarray):
|
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:
|
else:
|
||||||
img_w, img_h = img.size
|
img_w, img_h = img.size
|
||||||
tw, th = self.size
|
tw, th = self.size
|
||||||
|
Loading…
x
Reference in New Issue
Block a user