diff --git a/main.py b/main.py
index f7accf3e3a8e0c78a94f0ecc2c1d25dc0ee61c92..57ad95dfa84f49c2ad4909f207f0a3120b664faf 100644
--- a/main.py
+++ b/main.py
@@ -94,7 +94,7 @@ def eval_batch(eval_net, img_lst, crop_size=513, flip=True):
     batch_img = np.zeros((4, 3, crop_size, crop_size), dtype=np.float32)
     resize_hw = []
     for l in range(batch_size):
-        # img_ = img_lst[l]
+        img_ = img_lst[l]
         # resize_h, resize_w, _ = img_.shape
         # hwc to chw
         # img_ = img_.transpose((2, 0, 1))