FSA-FPN reconstruction method that fused self-attention mechanism based on YOLOX

Autor: An Henan, Guan Cong, Deng Wucai, Yang Jiazhou, Ma Chao
Jazyk: čínština
Rok vydání: 2023
Předmět:
Zdroj: Dianzi Jishu Yingyong, Vol 49, Iss 3, Pp 61-66 (2023)
Druh dokumentu: article
ISSN: 0258-7998
10490124
DOI: 10.16157/j.issn.0258-7998.223139
Popis: Abstract: With the increasing resolution of the input image of the current target detection task,the feature information extracted from the feature extraction network will become more and more limited under the condition that the receptive field of the feature extraction network remains unchanged,and the information coincidence degree between adjacent feature points will also become higher and higher.This paper proposes an FSA(fusion self-attention)-FPN,and designs SAU(self-attention upsample) module.The internal structure of SAU performs cross calculation with self-attention mechanism and CNN to further Feature fusion,and reconstructs FCU(feature coupling unit) to eliminate feature dislocation between them and bridge semantic gap. In this paper,a comparative experiment is carried out on Pascal VOC2007 data set using YOLOX-Darknet 53 as the main dry network. The experimental results show that compared with the FPN of the original network,the average accuracy of MAP@ [.5:.95] after replacing FSA-FPN is improved by 1.5%,and the position of the prediction box is also more accurate.It has better application value in detection scenarios requiring higher accuracy.
Databáze: Directory of Open Access Journals