fangliang425
fangliang425
> nms = True 谢谢回复, 代码没有修改过,centernet.py里的nms确实是True,模型也是链接的‘centernet_resnet50_voc.pth’。
> 那太奇怪了,我再试一下,我并没有修改过,你的测试集确认正确吗 测试集是VOC2007 test,并且全部用于测试,总共是4952个文件。运行你的yolov4-mobilenet,可以得到readme里的mAP。
> yolov4-mobilenet?这个不是centernet的库吗 > yolov4-mobilenet默认是mobilenet0.25本就这么低啊。人家参数就几十万。。 我的意思是用的相同的测试集在yolov4-mobilenet项目上能得到相应readme的结果,说明测试集是没问题的。 这个centernet项目里只能得到0.67 mAP,readme里的是0.77. 都是pytorch版本
> 我试了一下,还是77% > 关闭nms是62%,是不是版本不同会影响性能啊。我不太懂。。。 刚找了旧的GPU,试了下pytorch 1.60, 1.2.0,还有30系显卡上的1.70,都是一样0.67(nms为false 是0.44)的结果。。。请问你是把VOC2007_test里的所有sample都用于测试吗,总共4952个sample。(voc2centernet.py里trainval_percent=0, train_percent=1生成的) `72.91% = aeroplane AP || score_threhold=0.5 : F1=0.63 ; Recall=46.67% ; Precision=96.38% 75.09% = bicycle AP || score_threhold=0.5 : F1=0.60 ;...
问题找到了,原来我把resnet50 的backbone换成了pytorch官方的,导致一部分layer的weight无法载入。😂 感谢一直耐心的回复,也很欣赏这种细致的风格。已经star了
> > 问题找到了,原来我把resnet50 的backbone换成了pytorch官方的,导致一部分layer的weight无法载入。 > > 感谢一直耐心的回复,也很欣赏这种细致的风格。已经star了 > > 你好,我能看一下你“原来我把resnet50 的backbone换成了pytorch官方的,导致一部分layer的weight无法载入。joy” > 的错误代码吗? > 邮箱:[[email protected]](mailto:[email protected]) > 谢谢 并不会报错,部分weight无法的载入是我的推测,注释部分是载入官方resnet ```` def resnet50(pretrain = True): model = ResNet(Bottleneck, [3, 4, 6, 3])...