问题分析
最近在训练自己的数据集时,训练总是中断,报错提示
Expected image (JPEG, PNG, or GIF), got unknown format starting with '\000\000\000\001Bud1\000\000 \000\000\000\010\000'
根据报错信息提示是由于在使用Python语言读取本地文件的时候,图片文件夹中包含非jpeg,png,gif格式的文件,导致读取无效。
解决方案
只需将文件夹中的非jpeg,png,gif格式的文件删除就好。但是我在排查多次之后,依然未发现有其他文件,但每次训练依旧报错,最后灵机一动,尝试遍历读取文件夹中的每张图片,当读取非图片文件时会报错,最后进行删除,感觉蛋疼。。。。。
参考代码
import