昨天做好训练集之后运行train.py,结果一直提示“can not train without labels”.
尝试了诸如修改训练集路径等方法后依旧无果,于是参考大佬的数据集设置,发现jpg文件的命名跟txt的命名是一一对应的,而我在保存标签时都加上了“train”,导致运行时找不到对应的标签。
然后因为这种低级错误浪费了一天时间。
这里保存一下修改文件名用的代码 代码来源(@liguo_yao)
# -*- coding: utf-8 -*-
import os
#设定文件路径
path='G:\\whorl_nut\\'
i=1
#对目录下的文件进行遍历
for file in os.listdir(path):
#判