我的数据集中有的图片没有目标框,xml中也没有object,运行到voc.py中的color_print会报错:
Traceback (most recent call last):
File "train.py", line 19, in
from dataloader.dataloaders import train_dataloader, val_dataloader
File "D:\easy_detection\easy_detection\dataloader\dataloaders.py", line 25, in
train_dataset, val_dataset = get_all_datasets()
File "D:\easy_detection\easy_detection\dataloader\list_dataset.py", line 82, in get_all_datasets
first_gpu=is_first_gpu())
File "D:\easy_detection\easy_detection\dataloader\voc.py", line 160, in init
utils.color_print(f'{name}: {counter[name]} ({counter[name]/tot_bbox*100:.2f}%)', 5)
ZeroDivisionError: division by zero
注释掉这里的代码之后可以运行,但是输出结果全是nan
我的数据集中有的图片没有目标框,xml中也没有object,运行到voc.py中的color_print会报错:
Traceback (most recent call last):
File "train.py", line 19, in
from dataloader.dataloaders import train_dataloader, val_dataloader
File "D:\easy_detection\easy_detection\dataloader\dataloaders.py", line 25, in
train_dataset, val_dataset = get_all_datasets()
File "D:\easy_detection\easy_detection\dataloader\list_dataset.py", line 82, in get_all_datasets
first_gpu=is_first_gpu())
File "D:\easy_detection\easy_detection\dataloader\voc.py", line 160, in init
utils.color_print(f'{name}: {counter[name]} ({counter[name]/tot_bbox*100:.2f}%)', 5)
ZeroDivisionError: division by zero
注释掉这里的代码之后可以运行,但是输出结果全是nan