网络上普遍对于这个报错信息得出的问题为:标签分类错误、标签分类超限等等,我使用以下代码段对训练集和验证集进行脏数据搜索,发现数据没有问题。
并且使用了Inria数据集与Potsdam数据集进行语义分割,发现同样会在几个epoch后出现这个错误,并且用了pycharm与spyder两个不同的编译器,错误仍然存在,因此错误原因大概率在代码中。
import os
import PIL.Image as Image
import numpy as np
import cv2
# 遍历文件夹
def walkFile(file):
for root, dirs, files in os.walk(file):
# root 表示当前正在访问的文件夹路径
# dirs 表示该文件夹下的子目录名list
# files 表示该文件夹下的文件list
# 遍历文件
for f in files:
img = cv2.imread(os.path.join(root, f))
for j in range(256):
for k in range(256):
if (img[j][k][0] == 255 and img[j][k][1] == 0 and img[j][k][2] == 255) or (img[j][k][0] == 0 and img[j][k][1] == 0 and img[j][k][2] == 0) or (img[j][k][0] == 0 and img[j][k][1] == 0 and img[j][k][2] == 255) or (img[j][k][0] == 255 and im