opencv清除文件夾中空圖片或損壞圖片

import os
import cv2
import numpy as np

path = "/raid/data/jianhuang_test/jpg"
imgList = os.listdir(path)
for i in imgList:
    imgPath = os.path.join(path, i)
    if os.path.getsize(imgPath) == 0:
        os.remove(imgPath)
        continue
    image = cv2.imdecode(np.fromfile(imgPath, dtype=np.uint8), -1)
    try:
        image.shape
    except:
        os.remove(imgPath)