Research on YOLOX for small target detection in infrared aerial photography based on NAM channel attention mechanism

Xiangyang Li; Zhaofei Xu; Qing Liu; Wei Xue; Guanghui Yue; Shulun Wang

doi:10.1117/12.2663245

12 April 2023 Research on YOLOX for small target detection in infrared aerial photography based on NAM channel attention mechanism

Xiangyang Li, Zhaofei Xu, Qing Liu, Wei Xue, Guanghui Yue, Shulun Wang

Author Affiliations +

Proceedings Volume 12565, Conference on Infrared, Millimeter, Terahertz Waves and Applications (IMT2022); 125653T (2023) https://doi.org/10.1117/12.2663245
Event: Conference on Infrared, Millimeter, Terahertz Waves and Applications (IMT2022), 2022, Shanghai, China

Abstract

In this paper, we propose an improved target detection algorithm based on YOLOX with attention mechanism for infrared aerial photography, which is used to solve the problem that small targets are easily lost in the environment with dense targets and complex background. First of all, a new channel attention module named SA-NAM is introduced. This makes it possible to utilize the contribution factor of weights to optimize attentional mechanism and SA function to enhance the feature extraction ability in network. Target distribution in real scenarios has long tail distribution, we also adopted Mosaic and Cutmix data enhancement methods and Focal loss function to category imbalance. Moreover, in order to fill the gap of infrared small target data set of aerial views in current target detection field, an infrared aerial small target data set containing human, motor vehicle and non-motor vehicle targets in various scenarios is constructed. Experiments show that the mAP of proposed algorithm on our data set has reached 63.7%, and the detection accuracy is improved by 6.2% compared with baseline YOLOX, which alleviates the misdetection of infrared small target has been ameliorated to a certain extent.

Citation Download Citation

Xiangyang Li, Zhaofei Xu, Qing Liu, Wei Xue, Guanghui Yue, and Shulun Wang "Research on YOLOX for small target detection in infrared aerial photography based on NAM channel attention mechanism", Proc. SPIE 12565, Conference on Infrared, Millimeter, Terahertz Waves and Applications (IMT2022), 125653T (12 April 2023); https://doi.org/10.1117/12.2663245

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available