Skip to content

Commit 76429f4

Browse files
authored
Merge pull request #7744 from yeyupiaoling/my-cool-stuff
Fix the arguments transfer in image.py.
2 parents d43932c + fbda30a commit 76429f4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/paddle/v2/image.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -319,11 +319,11 @@ def simple_transform(im,
319319
"""
320320
im = resize_short(im, resize_size)
321321
if is_train:
322-
im = random_crop(im, crop_size)
322+
im = random_crop(im, crop_size, is_color=is_color)
323323
if np.random.randint(2) == 0:
324324
im = left_right_flip(im)
325325
else:
326-
im = center_crop(im, crop_size)
326+
im = center_crop(im, crop_size, is_color=is_color)
327327
if len(im.shape) == 3:
328328
im = to_chw(im)
329329

0 commit comments

Comments
 (0)