OSFormer: One-Step Transformer for Infrared Video Small Object Detection.

Journal: IEEE transactions on image processing : a publication of the IEEE Signal Processing Society
Published Date:

Abstract

Infrared video small object detection is pivotal in numerous security and surveillance applications. However, existing deep learning-based methods, which typically rely on a two-step paradigm of frame-by-frame detection followed by temporal refinement, struggle to effectively utilize temporal information. This is particularly challenging when detecting small objects against complex backgrounds. To address these issues, we introduce the One-Step Transformer (OSFormer), a novel method that pioneeringly integrates a small-object-friendly transformer with a one-step detection paradigm. Unlike traditional methods, OSFormer processes the video sequence only through a single inference, encoding the sequence into cube format data and tracking object motion trajectories. Additionally, we propose the Varied-Size Patch Attention (VPA) module, which generates patches of varying sizes to capture adaptive attention features, bridging the gap between transformer architectures and small object detection. To further enhance detection accuracy, OSFormer incorporates a Doppler Adaptive Filter, which integrates traditional filtering techniques into an end-to-end neural network to suppress background noise and accentuate small objects. OSFormer outperforms YOLOv8-s on both the AntiUAV dataset (+3.1% mAP, -35.1% Params) and the InfraredUAV dataset (+4.0% mAP, -51.0% FLOPs), demonstrating superior efficiency and effectiveness in small object detection.

Authors

  • Haolin Qin
  • Tingfa Xu
    Ministry of Education Key Laboratory of Photoelectronic Imaging Technology and System, School of Optics and Photonics, Beijing Institute of Technology, Beijing, China.
  • Yuan Tang
    School of Humanities and Arts, Jiaxing Nanhu University, Jiaxing 314000, Zhejiang, China.
  • Fengxiang Xu
  • Jianan Li
    Department of Cardiology, Fuwai Hospital, National Center for Cardiovascular Diseases, Chinese Academy of Medical Sciences and Peking Union Medical College, Beijing, PR China.

Keywords

No keywords available for this article.