科研之友
微信
新浪微博
Facebook
分享链接
摘要
对于两阶段目标检测算法中模型存在检测精度低、小目标漏检率高等问题，提出通道分离双注意力机制的目标检测算法，通过改进Faster+FPN主干网络来提高小目标的检测精度。首先针对神经网络不能自动学习特征间的重要性问题，在通道分离过程中提出双注意力机制来构建深度神经网络，另结合分组卷积、空洞卷积等技术减少网络参数。其次针对高分辨率特征经过深度CNN后导致的信息丢失问题，通过添加细节提取模块以及通道注意力特征融合模块来提取更多的细节特征。最后考虑到一般损失函数不可重点评估目标物位置的置信度，结合KL散度进行损失函数优化，通过训练使得预测分布更接近真实分布，有效地解决了神经网络直接用于目标检测存在的问题。采用PASCAL VOC2007、KITTI以及Pedestrian三类数据集对网络进行训练，并将提出的模型与多个目标检测算法进行对比。实验结果表明，该算法能够高效地对图像进行识别，且具有较高的检测精度。
- 单位
