The application of object detection technology has been used in people's daily lives widely, and it can be used accurately in traffic scenes is also crucial. Based on the YOLOX object detection model, a lightweight neural network for traffic object detection is proposed. First, we propose S-PANet (Simple Path Aggregation Network) structure based on the YOLOX_S baseline model. The purpose of the S-PANet structure is to reduce the parameters number of models and the amount of floating-point computation in training. Second, we propose to add CBAM (Convolutional Block Attention Module) to the network based on the S-PANet. Adding CBAM attention mechanism can improve the detection accuracy of models, but has less computation added. It is a compensation of the simplification on feature pyramid. Final, we carry out the lightweight experiments on three kinds of models, including S, Tiny, and Nano. The proposed methods are verified on KITTI and PASCAL VOC datasets. Our S model can achieve 92.9% mAP@0.5 and 21.33 FPS on KITTI dataset. The Tiny and Nano models can achieve 84.3%, 79.4% mAP@0.5 and 33.07, 41.67 FPS respectively. Compared with the baseline models, the detection accuracies mAP@0.5 have less difference and the parameters number of the models are reduced.


    Zugriff

    Zugriff prüfen

    Verfügbarkeit in meiner Bibliothek prüfen

    Bestellung bei Subito €


    Exportieren, teilen und zitieren



    Titel :

    A Lightweight Neural Network Model of Feature Pyramid and Attention Mechanism for Traffic Object Detection


    Beteiligte:
    Gao, Le-yuan (Autor:in) / Qu, Zhong (Autor:in) / Wang, Shi-yan (Autor:in) / Xia, Shu-fang (Autor:in)

    Erschienen in:

    Erscheinungsdatum :

    01.02.2024


    Format / Umfang :

    5966229 byte




    Medientyp :

    Aufsatz (Zeitschrift)


    Format :

    Elektronische Ressource


    Sprache :

    Englisch



    A DENSELY CONNECTED FEATURE PYRAMID NETWORK FOR OBJECT DETECTION

    Zhang, X. / Yan, L. / Gerada, C. | TIBKAT | 2021


    Lightweight Traffic Accident Detection Algorithm Based on Attention Mechanism

    Hou, XiaoYan / Wen, ShiHuan / Chen, ShengFu et al. | IEEE | 2024


    A lightweight network for traffic sign recognition based on multi-scale feature and attention mechanism

    Wei, Wei / Zhang, Lili / Yang, Kang et al. | Elsevier | 2024

    Freier Zugriff

    Research on Bidirectional Multi-Span Feature Pyramid and Key Feature Capture Object Detection Network

    Heng Zhang / Faming Shao / Xiaohui He et al. | DOAJ | 2024

    Freier Zugriff

    Road Traffic Vehicle Detection Method Using Lightweight YOLOv5 and Attention Mechanism

    Wang, Yunzhen / Ma, Hongbing / Li, Liangliang | IEEE | 2022