Python检测JPEG、PNG是否完整

《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))