Python检测数据不完整的JPG图片

王小鹏鹏

已于 2025-02-20 17:31:34 修改

阅读量6.7k

点赞数 8

分类专栏： OpenCV 文章标签： python

于 2019-01-29 16:47:43 首次发布

本文链接：https://blog.youkuaiyun.com/kingroc/article/details/86692156

版权

OpenCV 专栏收录该内容

78 篇文章

订阅专栏

本文介绍如何利用Python检测JPEG图片是否完整。通过查找文件尾部的标识来判断图片是否因下载不完全而损坏。代码示例提供了一个解决方案，并提醒注意删除不完整图片的代码行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

《OpenCV系列教程》
项目位置：OpenCV-Sample
代码位置：32-FindIncompleteJPG.py

《Python检测JPEG、PNG是否完整（2）》
有时候图片下载后就是这样的：
在这里插入图片描述
因为图片没有下载结束，仅仅显示了下载那一部分。网上方法很多，经过实践，我找到了一个较好的方法就是判断jpg文件结尾的标识。
代码如下：
记得打开os.remove(train_dir + file)这一句。

import os

train_dir = './res/'

def progress(percent, width=50):
    '''进度打印功能'''
    if percent >= 100:
        percent = 100

    show_str = ('[%%-%ds]' % width) % (int(width * percent / 100) * "#")  # 字符串拼接的嵌套使用
    print('\r%s %d%% ' % (show_str, percent), end='')

def is_valid_jpg(jpg_file):
    with open(jpg_file, 'rb') as f:
        f.seek(-2, 2)
        buf = f.read()
        f.close()
        return buf ==  b'\xff\xd9'  # 判定jpg是否包含结束字段

data_size = len([lists for lists in os.listdir(train_dir) if os.path.isfile(os.path.join(train_dir, lists))])
recv_size = 0
incompleteFile = 0
print('file tall : %d' % data_size)

for file in os.listdir(train_dir):
    if os.path.splitext(file)[1].lower() == '.jpg':
        ret = is_valid_jpg(train_dir + file)
        if ret == False:
            incompleteFile = incompleteFile + 1
            #os.remove(train_dir + file)

    recv_per = int(100 * recv_size / data_size)
    progress(recv_per, width=30)
    recv_size = recv_size + 1

progress(100, width=30)
print('\nincomplete file : %d' % incompleteFile)