出现该异常有很多种情况:
我说下我遇到的情况 希望对你有帮助
当模型运行fit_generator会抛出generator raised StopIteration的异常,其实是因为generator没有获取到相应的数据
这里的G是LoadBatches1.imageSegmentationGenerator做的一个生成器
G = LoadBatches1.imageSegmentationGenerator(train_images_path,train_segs_path, train_batch_size, n_classes=n_classes, input_height=input_height, input_width=input_width)
里边源代码读取图片和为cv2.imread 但是含有中文路径的图无法读取
解决方法 改为im=cv2.imdecode(np.fromfile(im, dtype=np.uint8),1)可以读取含有中文的文件路径
另一个问题是使用上述代码读取标签文件 只能读取8位的标签文件 但labelme制作出的标签文件为16位 这里涉及到代码转换
解决代码:
#读取uint16的标签文件 转化成uint8的标签文件
from skimage import io
seg1 = io.imread(seg) #Todo:使用skimage模块读取图片不会改变图片的数据格式
print(seg1.dtype)
seg1 = seg1.astype(np.uint8)
然后注意下文件路径的设置就好了,多测试可以解决的
有问题和建议欢迎下方留言!!!