centernet-pytorch icon indicating copy to clipboard operation
centernet-pytorch copied to clipboard

训练结果

Open xdnjust opened this issue 4 years ago • 8 comments
trafficstars

大神你好,跟着你代码学了很多东西,基本都star了遍 我想问一下你这个跑出来的COCO mAP比论文提供的结果好像要差点,这是啥原因啊

xdnjust avatar May 29 '21 13:05 xdnjust

大概差了多少?

bubbliiiing avatar May 30 '21 03:05 bubbliiiing

CenterNet-HG Hourglass-104,mAP 42.1 / 45.1,mAP50 61.1 / 63.9 (斜杠前后分别代表single-scale和multi-scale testing)

结果和你的有点差别,是否和训练epoch等调参有关?

xdnjust avatar May 31 '21 03:05 xdnjust

emm我搞错了,源库是40.3应该,42.1是有翻转的,我没有修改什么东西,权值用的都是官方的,原因我也不太了解0 0

bubbliiiing avatar Jun 05 '21 15:06 bubbliiiing

@bubbliiiing 工程里得centernet_resnet50_voc.pth文件是官方给的还是您自己训练出来的

chenpeng68 avatar Jul 07 '21 08:07 chenpeng68

训练的

bubbliiiing avatar Jul 15 '21 05:07 bubbliiiing

请问工程中的centernet_resnet50_voc.pth(mAP0.5=77.1),这个的训练方式是通过:在train.py中设置从resNet50主干的预训练权值开始训练开始训练吗?(即model_path = '' ; pretrained = True)

juAo12 avatar Nov 28 '21 07:11 juAo12

yes

bubbliiiing avatar Nov 28 '21 14:11 bubbliiiing

非常感谢

juAo12 avatar Nov 28 '21 14:11 juAo12