删除文件中不能打开的图片

在生成数据集的时候需要使用背景图片,但是从网上下载的背景图片有的不能读取,所以要把它删掉。

发现图片虽然不能打开,但是在inshow的时候会报错,于是用show做判断,能show出来就继续,不能就删除。

ubuntu+python2.7

还需要加一个

# -*- coding:utf-8 -*-

 1 import os
 2 import cv2
 3 
 4 #图片文件夹路径
 5 filedir = os.getcwd() + '/bg_img'
 6 #文件夹下所有图片的名字
 7 photonames = os.listdir(filedir)
 8 
 9 for photo in photonames:
10     #图片路径
11     photodir = filedir + '/' + photo
12     image = cv2.imread(photodir)
13     try:
14         cv2.imshow("img", image)
15     except:
16         os.remove(photodir)
17         continue

猜你喜欢

转载自www.cnblogs.com/j657521265/p/9379332.html