Fix Reader (#1285)

94002d7b · Chang Xu · GitHub · 968985e2 · 94002d7b
显示空白变更内容
内联并排

Showing with 6 addition and 1 deletion

demo/imagenet_reader.py demo/imagenet_reader.py +6 -1

未找到文件。
--- a/demo/imagenet_reader.py
+++ b/demo/imagenet_reader.py
@@ -99,7 +99,12 @@ def distort_color(img):
    return img


-def process_image(sample, mode, color_jitter, rotate, crop_size, resize_size):
+def process_image(sample,
+                  mode,
+                  color_jitter,
+                  rotate,
+                  crop_size=DATA_DIM,
+                  resize_size=RESIZE_DIM):
    img_path = sample[0]

    try: