diff --git a/demo/imagenet_reader.py b/demo/imagenet_reader.py
index 609bfba168520610f4749d0546a69f3bf321fa55..85abbb22872fae23dfe5ad9f75d9907b5cfb14d1 100644
--- 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: