fix decode bug #7741

992f20ce · yeyupiaoling · 29603cf3 · 992f20ce
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

python/paddle/v2/image.py python/paddle/v2/image.py +2 -2

未找到文件。
--- a/python/paddle/v2/image.py
+++ b/python/paddle/v2/image.py
@@ -319,11 +319,11 @@ def simple_transform(im,
    """
    im = resize_short(im, resize_size)
    if is_train:
-        im = random_crop(im, crop_size)
+        im = random_crop(im, crop_size, is_color=is_color)
        if np.random.randint(2) == 0:
            im = left_right_flip(im)
    else:
-        im = center_crop(im, crop_size)
+        im = center_crop(im, crop_size, is_color=is_color)
    if len(im.shape) == 3:
        im = to_chw(im)