读数据集时报错PIL.UnidentifiedImageError: cannot identify image file <_io.BytesIO object a

PIL.UnidentifiedImageError: cannot identify image file <_io.BytesIO object at 0x000001E16E54B6D0>
[[{{node PyFunc}}]]
[[IteratorGetNext]] [Op:__inference_train_function_2141]
Function call stack:
train_function
在这里插入图片描述
有读取不出来的图片,要找出这些坏图片,删掉

'''
此代码是查找出一个文件夹里面,
所有图片读取错误,再删除
'''
import os
import shutil
import warnings
import cv2
import io
from PIL import Image
warnings.filterwarnings("error", category=UserWarning)
base_dir = "C:/pycharm/Birds-Images-Dataset-master/train/骨顶鸡"#删除图片的根目录
i = 0
def is_read_successfully(file):
    try:
        imgFile = Image.open(file)#这个就是一个简单的打开成功与否
        return True
    except Exception:
        return False
for parent, dirs, files in os.walk(base_dir):#(root,dirs,files)
    for file in files:
        if not is_read_successfully(os.path.join(parent, file)):
            print(os.path.join(parent, file))
            #os.remove(os.path.join(parent, file)) #真正使用时,这一行要放开,自己一般习惯先跑一遍,没有错误了再删除,防止删错。
            i = i + 1
print(i)

找到错误图片
在这里插入图片描述

参考博客:https://www.136.la/jingpin/show-175978.html感谢!

  • 6
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 10
    评论
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值