python删除文件夹里损坏的图片

今天学习pytoch时发现有好多图片损坏了
报错如下： PIL.UnidentifiedImageError: cannot identify image file <_io.BufferedReader name=’./train/cat/052cat.jpg’>
原因
图片没有被正确的下载
解决
写个小程序把所有损坏的图片都删除
import cv2
import os
import numpy as np
fn_all = './test/cat/'
for fn in os.listdir(fn_all):
img = cv2.imread(fn_all + fn)
if type(img) != type(np.array([1])):
os.remove(fn_all + fn)

小程序改进：
import cv2
import os
import n

python删除文件夹里损坏的图片最先出现在Python成神之路。