标签:下载 class 判断 dir pytho col for cep 就删除
在生成数据集的时候需要使用背景图片,但是从网上下载的背景图片有的不能读取,所以要把它删掉。
发现图片虽然不能打开,但是在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
标签:下载 class 判断 dir pytho col for cep 就删除
原文地址:https://www.cnblogs.com/j657521265/p/9379332.html