python校验图片是否损坏【完全损坏或部分像素缺失】

文章目录


前言

图片数据训练时如果遇到图片损坏的情况可能会导致训练终止,可以加入这个方法进行处理


import os
import imghdr
from PIL import Image


def is_img_break(img_path):
    '''
    检查图片是否损坏:损坏返回True, 没坏返回False
    :param img_path: 图片路径
    :return:
    '''
    break_flag = False
    # 判断图片是否完全损坏

猜你喜欢

转载自blog.csdn.net/weixin_43721000/article/details/129851400