Ubuntu CenterNet训练自己的数据集(全过程 可训练可测试 附带整个工程文件包含数据集与修改后的代码)

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改成自己的

  • 2
    点赞
  • 20
    收藏
    觉得还不错? 一键收藏
  • 8
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值