提交 2c8ed668 编写于 作者: W wuzewu

Fix data augmentation bug in ImageClassificationReader

上级 aa4e1397
......@@ -112,7 +112,7 @@ class ImageClassificationReader(BaseReader):
image = Image.open(image_path)
image = image_augmentation.image_resize(image, self.image_width,
self.image_height)
if self.data_augmentation:
if self.data_augmentation and phase == "train":
image = image_augmentation.image_random_process(
image, enable_resize=False, enable_crop=False)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册