首先官方给的教程里面都是coco数据集的,如果我想要其他格式的自创数据集怎么办
参考
https://blog.csdn.net/laizi_laizi/article/details/104256781
https://zhuanlan.zhihu.com/p/630624425
首先啊改voc.py和改classname.py好像都要改,主要是为了符合数据集的标签划分
然后数据加载和网络这里,我看config/pascal_voc里面有
faster-rcnn_r50_fpn_1x_voc0712.py
然后它不是继承了
base = [
‘…/base/models/faster-rcnn_r50_fpn.py’, ‘…/base/datasets/voc0712.py’,
‘…/base/default_runtime.py’
]
这三个类吗?所以我们只要修改这三个类,或者把这三个类修改为我们需要的三个类,不就可以了吗?
所以按照那篇知乎文章的修改
得到了如下的结果:
需要注意的:
1、‘…/base/default_runtime.py’ 这个好像一定要继承,不能只用schedule_1x等?不然会出现这个问题 runner = Runner.from_cfg(cfg) ‘NoneType’ object has no attribute ‘get’
2、VOC.py改的时候不能改加注释’‘’ ‘’’ 直接删掉原来的类,不然会出现这个问题:AssertionError: classes
in XMLDataset
can not be None.