CenterNet:Objects as Points目标检测模型在Pytorch当中的实现
目录
性能情况
训练数据集
权值文件名称
测试数据集
输入图片大小
mAP 0.5:0.95
mAP 0.5
VOC07+12
VOC-Test07
512x512
-
77.1
COCO-Train2017
COCO-Val2017
512x512
38.4
56.8
所需环境
torch==1.2.0
注意事项
代码中的centernet_resnet50_voc.pth是使用voc数据集训练的。
代码中的centernet_hourglass_coco.pth是使用coco数据集训练的。
注意不要使用中文标签,文件夹中不要有空格!
在训练前需要务必在model_data下新建一个txt文档,文档中输入需要分的类,在train.py中将classes_path指向该文件。
文件下载
训练所需的
1