最近在训练的时候出现一个问题,而且老是同一个地方问题,OSError: IOError: broken data stream when reading image file。仔细的追踪代码,本以为路径有问题,结果发现都是没问题,后来分析才知道很大概率图像本身可能存在一些问题,经过网上大佬指点,最终解决如下:在py代码加上一下代码
from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True#遇到图像有截断问题自动处理。建议这条代码强烈加上,这是个PIL 的bug,如果不加上这个代码就会遇到错误就不执行Image相关打开处理代码。本人在工程遇到的天坑问题!!!

在训练图像数据时遇到OSError: IOError: broken data stream when reading image file错误。问题根源在于部分图像文件可能存在问题。通过在Python代码中引入PIL库的ImageFile模块,并设置LOAD_TRUNCATED_IMAGES为True,可以自动处理截断的图像文件,避免程序因图像问题而中断。这是一个PIL库已知的bug,添加相应代码能有效解决此类问题。

3043

被折叠的 条评论
为什么被折叠?



