fix bug in RandCropImageV2

pull/2086/head
HydrogenSulfate 2022-06-20 10:53:19 +08:00
parent 851e915e6a
commit 3b1247d069
1 changed files with 1 additions and 1 deletions

View File

@ -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