Caught FileNotFoundError in DataLoader worker process 0
一般是数据集有问题,比如图片和标注文件不匹配,导致找不到文件
AssertionError Caught AssertionError in DataLoader worker process 0
查看一下标注文件是不是有些图片没有boundingbox,要删掉
KeyError Caught KeyError in DataLoader worker process 0
查看标注文件是不是有的boundingbox名字不对,多个空格啥得
发生异常: KeyError 'height'
有部分标注文件中的height不存在
float division by zero
标注文件中的图片尺寸中的宽度或者长度为0了
解决方法:
对于标注文件有问题的情况,使用如下代码进行查找文件夹中所有文件是否存在某串字符
import os
text = "<name>2 </name>"
path = "D:/2020.10apple_data/VOCapple/VOC2012/Annotations"
flag = 0
lister = []
def getfiles(path):
# 将flag设置全局变量,防止递归查找时信息提示错误。
global flag
# 到给定的路径下
os.chdir(path)
# 返回一个列表,列表中的元素为path下的文件和子文件夹
files = os.listdir()
# 依次遍历
for file_name in files:
# 返回文件或子文件夹的绝对路径
abs_path = os.path.abspath(file_name)
# 如果为文件则进行查找
if os.path.isfile(abs_path):
f = open(file_name, "r")
# 在文件中找到要查到的字符串后进行打印,并将flag置为1
if text in f.read():
flag = 1
print(text + " found in ")
print(abs_path)
# else:
# print("在{}中没查到!".format(abs_path))
# lister.append(file_name)
# print(lister)
# 如果为文件夹则重复前面的动作
if os.path.isdir(abs_path):
getfiles(abs_path)
if flag == 0:
print(text + " not found! ")
return False
return True
# def delete_files(): #定义函数名称
# for foldName, subfolders, filenames in os.walk(path): #用os.walk方法取得path路径下的文件夹路径,子文件夹名,所有文件名
# for filename in filenames: #遍历列表下的所有文件名
# for data in lister:
# if filename == data: #当文件名不为“aaa.txt”时
# os.remove(os.path.join(foldName, filename)) #删除符合条件的文件
# print("{} deleted.".format(filename)) ##输出提示
getfiles(path)
# delete_files() #调用定义的函数,注意名称与定义的函数名一致
更多推荐
Caught FileNotFoundError in DataLoader worker process 0
发布评论