Due to the low resolution, high noise, and strong spatial correlation of thermal infrared images, an improved YOLOv7 algorithm is proposed to solve the problem of low accuracy in road traffic infrared target detection algorithms. Introducing CBAM attention mechanism in the backbone network and Neck network to enhance the ability to locate small targets; Replace the original SPPCSPC module with an ASPP module in the Neck network to improve the detection accuracy of images with lower resolutions; Replace the original CIoU loss function with the WIoU loss function to enhance the detection accuracy of weak and small targets. Experiments on the Chinese thermal infrared data set CTIR show that compared with the YOLOv7 algorithm, the detection accuracy mAP value of the proposed algorithm is increased by 3.4%, and the detection performance is improved.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    Object Detection Algorithm Based on YOLOv7 Traffic Infrared Image


    Contributors:
    Guo, Wei (author) / Tang, Sitao (author)


    Publication date :

    2023-10-11


    Size :

    2510952 byte




    Type of media :

    Conference paper


    Type of material :

    Electronic Resource


    Language :

    English




    Traffic Rule Violation Detection using Detectron2 and Yolov7

    Choudhari, Rutvik / Goel, Shubham / Patel, Yash et al. | IEEE | 2023



    Improved YOLOv7 Target Detection Algorithm Based on UAV Aerial Photography

    Zhen Bai / Xinbiao Pei / Zheng Qiao et al. | DOAJ | 2024

    Free access