目标检测网络——Centernet搭建
一.参考文献
1. 文献1 CenterNet:Keypoint Triplets for Object Detection
2. 文献2 Objects as points
二.搭建环境
1.创建虚拟环境
conda create -n name python==3.8
2.所需库和软件
软件:cuda和cudnn
安装和下载参考 cuda和cudnn
相关库:
库的版本供参考,有些库的版本太老不支持下载可适当升高版本。
三.数据集
使用voc07+12数据集
数据集链接:
链接: 百度网盘 请输入提取码
提取码: ph32
下载解压后得到
第一个文件下:图像的标签信息
第二个文件下:对训练集,测试集,验证集的划分
第三个文件下:图像,共21504张
四.网络代码
源码:github
训练的主程序
数据集划分程序
目前程序还在训练,已经训练了10代,结果为