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