用于检查损坏的图像数量的 Python 脚本 - Python

如果你需要快速检查大量的图像数据集是否损坏，那么这个 Python 脚本可以帮助你。

使用方法

1. 安装所需的 Python 库

在终端执行以下命令:

pip install pillow

2. 下载脚本并将其保存到你的工作目录中

可以在以下链接中找到脚本文件：https://github.com/example/script.py

3. 打开终端并进入工作目录

cd path/to/your/work/directory

4. 运行脚本

python script.py

注意：默认情况下，脚本将检查与脚本文件在同一目录中的所有 JPG 图像。你可以在脚本中更改图像类型或指定其他目录中的图像文件。

脚本代码

下面是 Python 脚本的代码：

import os
from PIL import Image

# 设置要检查的图像类型
image_types = ['.jpg']

# 遍历当前目录中的所有文件和目录
for root, dirs, files in os.walk('.'):
    for filename in files:
        # 检查图像文件
        if any(filename.lower().endswith(image) for image in image_types):
            try:
                # 打开图像文件并检查它是否完整
                with Image.open(os.path.join(root, filename)) as img:
                    img.verify()
            except (IOError, SyntaxError) as e:
                # 图像文件已损坏
                print('Bad file: {}'.format(os.path.join(root, filename)))

结束。现在你应该可以使用这个脚本来检查你的图像数据集中是否有损坏的图像文件。