search

bubbliiiing / centernet-pytorch

这是一个centernet-pytorch的源码，可以用于训练自己的模型。
MIT License
366 stars 98 forks source link

我将输入改为416X416，ap结果有个别类很差，map也很差 #33

Open tty0013 opened 2 years ago

tty0013 commented 2 years ago

image image image image image 同样的数据集，我在ultralytics版yolov3中训练，结果好很多，请问怎么排查问题呢？

bubbliiiing commented 2 years ago

…我还特意看了一下，你为啥要拿centernet和yolo3比……难道不应该拿我的yolo3和他的yolo3比吗…

tty0013 commented 2 years ago

…我还特意看了一下，你为啥要拿centernet和yolo3比……难道不应该拿我的yolo3和他的yolo3比吗…

不好意思，可能我表述的不严谨，就是这个centernet的性能有点低于我的心理预期了。从损失函数的图像来看，已经收敛了，训练更多的解冻轮次，效果会更好吗？

bubbliiiing commented 2 years ago

centernet对有些目标是很弱，你看过centernet论文么，resnet50的backbone就20多的map

tty0013 commented 2 years ago

centernet对有些目标是很弱，你看过centernet论文么，resnet50的backbone就20多的map

好的，谢谢。

bubbliiiing commented 2 years ago

你改小input_shape之后应该会更弱……建议还是换一个网络，只拿它对比可以，1800张图片 200代还是少了一些

tty0013 commented 2 years ago

你改小input_shape之后应该会更弱……建议还是换一个网络，只拿它对比可以，1800张图片 200代还是少了一些

好的，我换成hourglass在试试

lkhwhkzz commented 11 months ago

你改小input_shape之后应该会更弱……建议还是换一个网络，只拿它对比可以，1800张图片 200代还是少了一些

好的，我换成hourglass在试试

看nms有没有开启