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

---

前言

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

---

import os
import imghdr
from PIL import Image


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