Laser & Infrared, Volume. 54, Issue 9, 1469(2024)
Research on infrared small target detection based on improved CenterNet
With the continuous development of machine learning technology, the research on object detection technology is becoming increasingly popular. To address the issues of low accuracy and poor real-time performance in target detection, a single stage object detection algorithm CenterNet is adopted to achieve rapid recognition of targets. A CBAM attention mechanism is added to resnet50, the backbone network of the algorithm, to improve the recognition accuracy of the network on the target. In the output module of the network, a new GSConv convolution module is used to improve the detection speed without loss of accuracy. The improved algorithm is validated on the infrared datasetand its detection accuracy reaches 82.91%. The results show that that the improved CenterNet algorithm can accurately and efficiently accomplish the recognition of small infrared targets.
Get Citation
Copy Citation Text
NI An-qing, LI Jun, WANG Yao-hong. Research on infrared small target detection based on improved CenterNet[J]. Laser & Infrared, 2024, 54(9): 1469
Category:
Received: Nov. 29, 2023
Accepted: Apr. 30, 2025
Published Online: Apr. 30, 2025
The Author Email: LI Jun (cqleejun@163.com)