diff --git a/main.py b/main.py
index 315bb6093e9ff522eeb718755e505752c9c91e55..61974c816878eac283350e5cc37ec1400402d85f 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 = os.path.join('./datasets/JPEGImages/', line+'.bmp')
-        msk_path = os.path.join('./datasets/SegmentationClass/', line+'.png')
+        img_path = os.path.join('./datasets/JPEGImages/', line) + '.bmp'
+        msk_path = os.path.join('./datasets/SegmentationClass/', line) + '.png'
         img_ = cv2.imread(img_path)
         msk_ = cv2.imread(msk_path, cv2.IMREAD_GRAYSCALE)
         print(img_path)