diff --git a/main.py b/main.py
index 34a5a6e36c8806714369c292f78d19dc36a7ee6b..4fda33f34d343799d8b585d5862e62c46ad64862 100644
--- a/main.py
+++ b/main.py
@@ -147,8 +147,8 @@ def net_eval():
     bi = 0
     image_num = 0
     for i, line in enumerate(img_lst):
-        img_path = './datasets/JPEGImages/' + line + '.bmp'
-        msk_path = './datasets/SegmentationClass/' + line + '.png'
+        img_path = './datasets/JPEGImages/' + line.replace('\n', '') + '.bmp'
+        msk_path = './datasets/SegmentationClass/' + line.replace('\n', '') + '.png'
         img_ = cv2.imread(img_path)
         msk_ = cv2.imread(msk_path, cv2.IMREAD_GRAYSCALE)
         print(img_path)