李志伟,
张天琪
1.重庆邮电大学通信与信息工程学院 重庆 400065
2.信号与信息处理重庆市重点实验室 重庆 400065
基金项目:国家自然科学基金(61671095)
详细信息
作者简介:赵辉：女，1980年生，教授，博士生导师，研究方向为深空通信、信号理论与信息处理、图像信息处理
李志伟：男，1996年生，硕士，研究方向为计算机视觉、深度学习、目标检测
张天琪：男，1971年生，教授，博士生导师，研究方向为无线通信的智能信号处理、通信抗干扰和信息对抗
通讯作者:赵辉 zhaohui@cqupt.edu.cn
中图分类号:TN911.73; TP391.41计量
文章访问数:330
HTML全文浏览量:105
PDF下载量:92
被引次数:0
出版历程
收稿日期:2020-04-24
修回日期:2021-02-15
网络出版日期:2021-03-31
刊出日期:2021-07-10
Attention Based Single Shot Multibox Detector
Hui ZHAO,,Zhiwei LI,
Tianqi ZHANG
1. School of Communication and Information Engineering, Chongqing University of Posts and Telecommunications, Chongqing 400065, China
2. Chongqing Key Laboratory of Signal and Information Processing, Chongqing 400065, China
Funds:The National Natural Science Foundation of China (61671095)
摘要
摘要:单发多框检测器SSD是一种在简单、快速和准确性之间有着较好平衡的目标检测器算法。SSD网络结构中检测层单一的利用方式使得特征信息利用不充分，将导致小目标检测不够鲁棒。该文提出一种基于注意力机制的单发多框检测器算法ASSD。ASSD算法首先利用提出的双向特征融合模块进行特征信息融合以获取包含丰富细节和语义信息的特征层，然后利用提出的联合注意力单元进一步挖掘重点特征信息进而指导模型优化。最后，公共数据集上进行的一系列相关实验表明ASSD算法有效提高了传统SSD算法的检测精度，尤其适用于小目标检测。
关键词:目标检测/
注意力机制/
特征融合/
单发多框检测器
Abstract:Single Shot multibox Detector (SSD) is a object detection algorithm that provides the optimal trade-off among simplicity, speed and accuracy. The single use of detection layers in SSD network structure makes the feature information not fully utilized, which will lead to the small object detection are not robust enough. In this paper, an Attention based Single Shot multibox Detector (ASSD) is proposed. The ASSD algorithm first uses the proposed two-way feature fusion module to fuse the feature information to obtain the feature layer which containing rich details and semantic information. Then, the proposed joint attention unit is used to mine further the key feature information to guide the model optimization. Finally, a series of experiments on the common data set show that the ASSD algorithm effectively improves the detection accuracy of conventional SSD algorithm, especially for small object detection.
Key words:Object detection/
Attention mechanism/
Feature fusion/
Single Shot multibox Detector (SSD)
PDF全文下载地址:
https://jeit.ac.cn/article/exportPdf?id=f5f37080-bb8a-4549-896d-62499f0b386d
