CenterNet训练与测试文档说明
一. 训练过程:
整个过程主要参考 https://blog.csdn.net/weixin_41765699/article/details/100118353
若有不懂可以再结合看看。
整个工程包括数据集及代码都打包进了百度云,链接:https://pan.baidu.com/s/1oCIrTX0ecCUrGqb9nX_9Ug 提取码:r1a9
(1) 首先准备好VOC格式的数据集,利用VOC2COCO.py文件进行转换,注意事项如下:
xml_data_path 是VOC数据集根路径,其余三个json文件名为coco训练需要的格式。
这里的文件路径需要改成 最终的coco训练的路径,自己需要看情况调整。
(2)之后将生成的三个json文件放到 /CenterNet-master/data/Your DATA/annotations文件夹中,将所有图片放到/CenterNet-master/data/Your DATA/images文件夹中
(3)在src/lib/datasets/dataset里面新建一个“ Your DATA. py”,文件内容照着文件夹下coco.py改成自己的