Object Detection Algorithm Based on Multiheaded Attention

被引:3
|
作者
Jiang, Jie [1 ]
Xu, Hui [1 ]
Zhang, Shichang [2 ]
Fang, Yujie [1 ]
机构
[1] Natl Univ Def Technol, Coll Syst Engn, Changsha 410073, Hunan, Peoples R China
[2] Ocean Univ China, Sch Informat Sci & Engn, Songling Rd 238, Qingdao 266100, Shandong, Peoples R China
来源
APPLIED SCIENCES-BASEL | 2019年 / 9卷 / 09期
基金
中国国家自然科学基金;
关键词
object detection; attention mechanism; multiheaded attention;
D O I
10.3390/app9091829
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
This study proposes a multiheaded object detection algorithm referred to as MANet. The main purpose of the study is to integrate feature layers of different scales based on the attention mechanism and to enhance contextual connections. To achieve this, we first replaced the feed-forward base network of the single-shot detector with the ResNet-101 (inspired by the Deconvolutional Single-Shot Detector) and then applied linear interpolation and the attention mechanism. The information of the feature layers at different scales was fused to improve the accuracy of target detection. The primary contributions of this study are the propositions of (a) a fusion attention mechanism, and (b) a multiheaded attention fusion method. Our final MANet detector model effectively unifies the feature information among the feature layers at different scales, thus enabling it to detect objects with different sizes and with higher precision. We used the 512 x 512 input MANet (the backbone is ResNet-101) to obtain a mean accuracy of 82.7% based on the PASCAL visual object class 2007 test. These results demonstrated that our proposed method yielded better accuracy than those provided by the conventional Single-shot detector (SSD) and other advanced detectors.
引用
收藏
页数:15
相关论文
共 50 条
  • [1] Object detection algorithm based on attention mask fusion
    Dong Xiao-xiao
    He Xiao-hai
    Wu Xiao-hong
    Qing Lin-bo
    Teng Qi-zhi
    [J]. CHINESE JOURNAL OF LIQUID CRYSTALS AND DISPLAYS, 2019, 34 (08) : 825 - 833
  • [2] Novel infrared object detection and tracking algorithm based on visual attention
    Liu, Lei
    Chen, Xu
    Xia, Qi
    [J]. TARGET AND BACKGROUND SIGNATURES IV, 2018, 10794
  • [3] Camouflage Object Detection Algorithm Based on Edge Attention and Reverse Orientation
    He, Wenhao
    Ge, Haibo
    Cheng, Mengyang
    An, Yu
    Ma, Sai
    [J]. Computer Engineering and Applications, 2024, 60 (07) : 229 - 237
  • [4] Attention Mechanism-Based Object Detection Algorithm in Aerial Images
    Bai, Zongbao
    Zhang, Junju
    Gao, Yuan
    Hu, Youcheng
    [J]. LASER & OPTOELECTRONICS PROGRESS, 2023, 60 (12)
  • [5] SMALL OBJECT DETECTION ALGORITHM BASED ON CONTEXT INFORMATION AND ATTENTION MECHANISM
    Zhong Hang
    Li Fan
    Kuang Ping
    Gu Xiaofeng
    He Mingyun
    Tang Heng
    [J]. 2022 19TH INTERNATIONAL COMPUTER CONFERENCE ON WAVELET ACTIVE MEDIA TECHNOLOGY AND INFORMATION PROCESSING (ICCWAMTIP), 2022,
  • [6] Lightweight sandy vegetation object detection algorithm based on attention mechanism
    Hua, Zhongwei
    Guan, Min
    [J]. JOURNAL OF AGRICULTURAL ENGINEERING, 2023, 54 (01):
  • [7] Underwater object detection algorithm based on channel attention and feature fusion
    Zhang Y.
    Li X.
    Sun Y.
    Liu S.
    [J]. Xibei Gongye Daxue Xuebao/Journal of Northwestern Polytechnical University, 2022, 40 (02): : 433 - 441
  • [8] Object Detection Algorithm Based on Context Information and Self-Attention Mechanism
    Liang, Hong
    Zhou, Hui
    Zhang, Qian
    Wu, Ting
    [J]. SYMMETRY-BASEL, 2022, 14 (05):
  • [9] Salient Object Detection Algorithm Based on Dual-Attention Recurrent Convolution
    Xie Xueli
    Li Chuanxiang
    Yang Xiaogang
    Xi Jianxiang
    [J]. ACTA OPTICA SINICA, 2019, 39 (09)
  • [10] A Dense Small Object Detection Algorithm Based on a Global Normalization Attention Mechanism
    Wu, Huixin
    Zhu, Yang
    Wang, Liuyi
    [J]. APPLIED SCIENCES-BASEL, 2023, 13 (21):