模型训练出现UnidentifiedImageError: cannot identify image file ＜io.BytesIO object at 0x7faaa82bdb90＞

醉公子～

于 2021-06-16 17:12:45 发布

阅读量8.4k

点赞数 11

分类专栏：深度学习文章标签：神经网络 tensorflow python 图片

本文链接：https://blog.youkuaiyun.com/qq_44936246/article/details/117962404

版权

在使用Tensorflow进行深度学习训练时，将图片数据转为tfrecord格式过程中遇到UnidentifiedImageError，问题源于文件夹中包含.DS_Store系统文件和其他非图像文件。通过循环遍历并删除非图像文件的代码方法，可以有效解决这个问题。本文提供了针对复杂目录的图片数据集清理方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述
UnidentifiedImageError: cannot identify image file <io.BytesIO object at 0x7faaa82bdb90>

在跑深度学习tensorflow框架是制作图片格式向tfrecord格式转化中，读取文件夹中的图片出现系统文件时候报错，提示有.DS_Store系统文件，还有就是有其他类型的图片

html
png
bmp

反正什么乱七八糟的图片都有一大堆，特别是在数据集比较大的时候，3/4G的时候，特别头疼;

在这里我是使用循环遍历的方法找到“它”，此它非比它，就是它，找到他，删除它，哈哈哈

废话不多说直接上代码：

####################3
此代码是查找出一个文件夹里面，
所有图片读取错误，再删除
###############
import os
import shutil
import warnings
import cv2
import io
from PIL import Image
warnings.filterwarnings("error", category=UserWarning)
base_dir = "../../signs/train/dragon"#删除图片的根目录
i = 0
def is_read_successfully(file):
    try:
        imgFile

最低0.47元/天解锁文章