Fix Reader (#1286)

0384aa23 · Chang Xu · GitHub · b84cceb5 · 0384aa23
隐藏空白更改
内联并排

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: