- 数据集准备:制作VOC格式数据集
- 使用官方划分文件划分种子
官方代码 - 根据自己的需求修改划分代码:
由于我只制作了VOC2007的,所以删掉了关于2012的
同时由于我的xml文件里的folder不是当前的文件夹,所以做了适当修改
(暂时还没试微调能不能用这个种子,所以并不确定是否正确,当前正在跑base)
year = 'VOC2007' # tree.find("folder").text
- 修改data下代码:
1)builtin_meta.py文件
这部分修改为你自己的类别,由于不想改太多,所以3个split我全设置为一样的了。
2)builtin.py文件,由于只有2007的,所以这里只删掉了2012相关的部分
- 修改配置文件
修改类别,如果data只有2007的,需要删掉2012的
改完之后根据操作训练就好,base训练可以成功,微调还未尝试