DATransNet: Dynamic Attention Transformer Network for Infrared Small Target Detection
About
Infrared small target detection (ISTD) is widely used in civilian and military applications. However, ISTD encounters several challenges, including the tendency for small and dim targets to be obscured by complex backgrounds. To address this issue, we propose the Dynamic Attention Transformer Network (DATransNet), which aims to extract and preserve detailed information vital for small targets. DATransNet employs the Dynamic Attention Transformer (DATrans), simulating central difference convolutions (CDC) to extract gradient features. Furthermore, we propose a global feature extraction module (GFEM) that offers a comprehensive perspective to prevent the network from focusing solely on details while neglecting the global information. We compare the network with state-of-the-art (SOTA) approaches and demonstrate that our method performs effectively. Our source code is available at https://github.com/greekinRoma/DATransNet.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Infrared Small Target Segmentation | IRSTD-1K | IoU62.4 | 10 | |
| Infrared Small Target Segmentation | SIRST | mIoU63.5 | 10 |