Python3.7出现RuntimeError: generator raised StopIteration异常

出现该异常有很多种情况:

我说下我遇到的情况 希望对你有帮助

当模型运行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)

然后注意下文件路径的设置就好了,多测试可以解决的 

有问题和建议欢迎下方留言!!!

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值