python批量删除错误图片（jpg、png），并对图片按顺序重命名

人参海海

已于 2022-08-08 09:28:25 修改

阅读量2.1k

点赞数 3

文章标签： python 开发语言

于 2022-08-07 14:31:37 首次发布

本文链接：https://blog.csdn.net/m0_58954450/article/details/126210589

版权

这篇博客介绍如何利用Python处理一个包含错误图片的数据集，包括删除头部错误和数据错误的图片，并对整个图片集进行顺序重命名。适合于处理大量图片的场景。

摘要由CSDN通过智能技术生成

因为研究需要，下载了一个公开的亚洲人脸数据集，由于网络波动等问题，下载的图片出现了头部错误以及数据错误，因为数据集太过庞大，于是想到用python来实现错误数据删除以及顺序重命名。

代码如下：

"""
检测图片完整性
"""
import os
from tqdm import tqdm

class CheckImage(object):

    def __init__(self, img):
        with open(img, "rb") as f:
            # f.seek(-2, 2)
            self.img_text = f.read()
            f.close()

    def check_jpg_jpeg(self):
        """检测jpg图片完整性，完整返回True，不完整返回False"""
        buf = self.img_text
        return buf.endswith(b'\xff\xd9')

    def check_png(self):
        """检测png图片完整性，完整返回True，不完整返回False"""

        buf = self.img_text
        return buf.endswith(b'\xaeB`\x82')

"""
检测某个文件夹中不完整的图片，并删除
"""

class CheckBrockImage(object):
    def __init__(self, train_dir):
        self.train_di