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

Fix data augmentation bug in ImageClassificationReader

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