search

bubbliiiing / efficientdet-pytorch

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

博主您好，我才使用您的efficientDet代码时候，分别用了两个不同的数据集，但是效果却相差较大。 #20

Open BieberZhang opened 3 years ago

BieberZhang commented 3 years ago 
  这两个数据集的图像尺寸均在1000像素左右
   效果差的那个数据集，存在这样的情况：共有10类目标，我的训练策略是使用efficientd-4作为初始权重。所测试的结果其map仅能达到75左右，但是用frcnn ssd retinanet等均能达到85-89左右。但是当我改用efficient d-3作为初始权重的时候，在batchsize=4的设置下，经过100epoch所测试的的各类ap只有个位数大小，实在好奇怪。
  但是另一个数据集，效果很好可以达到非常理想的map，与frcnn ssd retinanet等均相近甚至更高。
 希望能够得到您的帮助！非常感谢！
bubbliiiing commented 3 years ago

我个人觉得是d4难收敛，如果你连ssd都能到85的话，那建议你用d0。