情况描述:
- 在训练代码中修改数据集读取路径,运行 train.py 文件 产生报错:
解决:
-
经过网上搜索借鉴以及查看,发现是数据集(训练集/验证集)文件夹中的图片命名中存在空格,导致数据列表读取为空,本质上是数据集的路径问题。
-
训练时,要仔细检查数据的路径读取,可以在代码中 print 出来相关参数,以验证查看。
-
若出现其他报错:
-
Caught FileNotFoundError in DataLoader worker process 0
一般也是数据集有问题,如图片和标注文件不匹配,导致找不到文件
- KeyError Caught KeyError in DataLoader worker process 0
查看标注文件是不是存在 bbox 名字不对,或者命名中存在空格
相关链接推送: