Infrared Technology, Volume. 46, Issue 12, 1371(2024)

Nighttime Object Detection in Infrared and Visible Images Based on Multi-Attention Mechanism

Ruihong LI1, Zhitao FU1、*, Shaochen ZHANG1, Jian ZHANG1, and Leiguang WANG2
Author Affiliations
  • 1Faculty of Land Resources Engineering, Kunming University of Science and Technology, Kunming 650093, China
  • 2Key Laboratory of State Forestry and Grassland Administration on Forestry and Ecological Big Data, Southwest Forestry University, Kunming 650024, China
  • show less

    Object detection has long been a research hotspot in the field of computer vision, and the YOLO series of object detection models is widely used in numerous fields. However, most current image data for object detection are based on a single type of sensor, which makes it difficult to fully characterize the imaging scene. The detected objects contain limited useful information, especially under conditions of low illumination, night, rain, and fog. To improve nighttime object detection, our study proposed a multi-attention mechanism for infrared and visible images. This mechanism combines the CBAM attention mechanism with a Transformer to obtain rich local and contextual information and reduce false detections. To verify the effectiveness of the method, five current mainstream object detection algorithms were selected and tested on a public infrared object detection dataset. The mAP of the proposed method improved from 62.6% to 71.5% compared to the original YOLOv7. This study also produced an infrared–visible fusion dataset for nighttime object detection. On this dataset, the mAP improved significantly from 79.90% to 94.80% compared to the original YOLOv7.

    Tools

    Get Citation

    Copy Citation Text

    LI Ruihong, FU Zhitao, ZHANG Shaochen, ZHANG Jian, WANG Leiguang. Nighttime Object Detection in Infrared and Visible Images Based on Multi-Attention Mechanism[J]. Infrared Technology, 2024, 46(12): 1371

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Jul. 23, 2023

    Accepted: Jan. 14, 2025

    Published Online: Jan. 14, 2025

    The Author Email: Zhitao FU (zhitaofu@126.com)

    DOI:

    Topics