【TensorFlow学习笔记】问题篇— —Python读取本地文件时报错Expected image (JPEG, PNG, or GIF)的解决方案

问题分析

最近在训练自己的数据集时,训练总是中断,报错提示

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 os
import tensorflow as tf
from PIL import Image
#图片保存地址
path = "D:/picture/"
num = 1  # 名称变量
for file in os.listdir(path + 'A'):
    image = Image.open("D:/pictur/A/"+ file)
for file in os.listdir(path + 'B'):
    image = Image.open("D:/pictur/B/"+ file)
    
发布了28 篇原创文章 · 获赞 2 · 访问量 2817

猜你喜欢

转载自blog.csdn.net/Jarvis_lele/article/details/104591649