python读图出错-数据清洗

最新推荐文章于 2025-03-12 19:10:00 发布

原创最新推荐文章于 2025-03-12 19:10:00 发布 · 583 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言 #后端

这篇博客讨论了在使用Python处理图像时遇到的读取错误，可能原因是Python内置图像处理库的版本不兼容。作者提出了解决方案，即通过检查图片是否能成功读取，如果不能则直接删除。代码示例展示了如何遍历文件夹，检查并移除无法读取的图像。

python读取图像会发生图像读取错误，一个说法是python自带读取图像的包版本不兼容导致。有时间还是要研究一下。没时间就简单粗暴点，直接把无法读取的图像清洗掉。

import cv2
import io

from PIL import Image

warnings.filterwarnings("error", category=UserWarning)

data_dir = "./data/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):
    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)