修改net_eval函数

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)