![d4c3635c9cca420a097dd8cc94dc2925.png](https://img-blog.csdnimg.cn/img_convert/d4c3635c9cca420a097dd8cc94dc2925.png)
本文首发于知乎专栏“人工智能从入门到逆天杀神”,本文以及本专栏所有算法源代码都可以在神力AI平台获取,如果你没有GPU但需要预训练模型或者你想获取更多开箱即用的AI算法,欢迎加入我们的会员,一杯咖啡即可带你入门AI,传送门: http:// manaai.cn 。本文欢迎二次转载,但请保留原作版权信息,即该段信息,谢谢合作!
在上一篇文章中,我给大家介绍了如何采用更加先进的特征抽取方式来做目标检测器的回归。于是我们尝试了如何构建一个FPN即特征金字塔。我们在上一篇文章结尾的时候,基本上构建完了所有的代码,但还没有训练并进行测试查看最终结果。
我们由于时间关系仅仅训练了4-5个Epoch,但效果已经非常不错,这一点跟SSD改进很大,任何一阶段算法诸如SSD,DSSD,RefineDet,RFBNet在训练的早期都会出现大量检测结果的情况,这得益于特征金字塔的bottom-up, stop-down的编解码策略以及focal loss的强大选择能力,在训练的早期就可以干掉大部分的背景。具体效果如何:
only 4 epochs result, about 30 epochs the result could be much more accurate: