ACE-STDN: An infrared small target detection network with adaptive contrast enhancement
Author:
Affiliation:

1.School of Information Science and Technology, ShanghaiTech University, Shanghai 201210, China;2.Key Laboratory of Infrared System Detection and Imaging Technology, Shanghai Institute of Technical Physics, Chinese Academy of Sciences, Shanghai 200083, China

Clc Number:

TP391.4

Fund Project:

Supported by the Youth Innovation Promotion Association CAS (2014216); Supported by the National Pre-research Program during the 14th Five-Year Plan (514010405).

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    Due to the long distance and complex background, it is hard for the infrared detecting and tracking system to find and locate the dim-small targets in time. The proposed method, ACE-STDN, aims to tackle this difficult task and improve the detection accuracy. First of all, an adaptive contrast enhancement subnetwork preprocesses the input infrared image, which is conducive for the low-contrast dim targets. Next, a detection subnetwork with a hybrid backbone takes advantage of both convolution and self-attention mechanisms. Besides, the regression loss is designed based on 2D Gaussian distribution representation instead of Intersection over Union measurement. To verify the effectiveness and efficiency of our method, we conduct extensive experiments on two public infrared small target datasets. The experimental results show that the model trained by our method has a significant improvement in detection accuracy compared with other traditional and data-based algorithms, with the average precision reaching 93.76%. In addition, ACE-STDN achieves outstanding detection performance in a multiclass object dataset and a general small object dataset, verifying the effectiveness and robustness.

    Reference
    Related
    Cited by
Get Citation

YE Xin-Yi, GAO Si-Li, Li Fan-Ming. ACE-STDN: An infrared small target detection network with adaptive contrast enhancement[J]. Journal of Infrared and Millimeter Waves,2023,42(5):701~710

Copy
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:December 19,2022
  • Revised:August 14,2023
  • Adopted:February 02,2023
  • Online: August 06,2023
  • Published: