Python检测JPEG、PNG是否完整

最新推荐文章于 2023-11-21 15:24:19 发布

原创最新推荐文章于 2023-11-21 15:24:19 发布 · 2.8k 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

Python 专栏收录该内容

26 篇文章

订阅专栏

本文介绍了一个使用Python脚本检查JPEG和PNG图片文件完整性的方法。通过读取文件末尾的特定字节来判断图片是否损坏，适用于OpenCV项目中的数据集预处理阶段。

《OpenCV系列教程》
《深度学习-如何提高数据集质量》

项目位置：OpenCV-Sample
代码位置：35-CheckJPEGPNG.py

之前检测过JPEG文件是否完整，这次把PNG也添加进来，检测方式还是比较粗略，不能覆盖的很广。

import os

path = './res/'

def is_valid_jpg(jpg_file):
    with open(jpg_file, 'rb') as f:
        f.seek(-2, 2)
        buf = f.read()
        return buf ==  b'\xff\xd9'

def is_valid_png(png_file):
    with open(png_file, 'rb') as f:
        f.seek(-2, 2)
        buf = f.read()
        return buf == b'\x60\x82'

for file in os.listdir('res'):
    pic_file = path + file
    isJpg = is_valid_jpg(pic_file)
    isPng = is_valid_png(pic_file)


    print("jpeg : %s, png %s, file %s " % (isJpg, isPng, file))