提交 3b1247d0 编写于 作者: H HydrogenSulfate

fix bug in RandCropImageV2

上级 851e915e
...@@ -430,7 +430,7 @@ class RandCropImageV2(object): ...@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册