IDEAS home Printed from https://ideas.repec.org/a/plo/pone00/0285654.html
   My bibliography  Save this article

An object detection algorithm combining self-attention and YOLOv4 in traffic scene

Author

Listed:
  • Kewei Lu
  • Fengkui Zhao
  • Xiaomei Xu
  • Yong Zhang

Abstract

Automobile intelligence is the trend for modern automobiles, of which environment perception is the key technology of intelligent automobile research. For autonomous vehicles, the detection of object information, such as vehicles and pedestrians in traffic scenes is crucial to improving driving safety. However, in the actual traffic scene, there are many special conditions such as object occlusion, small objects, and bad weather, which will affect the accuracy of object detection. In this research, the SwinT-YOLOv4 algorithm is proposed for detecting objects in traffic scenes, which is based on the YOLOv4 algorithm. Compared with a Convolutional neural network (CNN), the vision transformer is more powerful at extracting vision features of objects in the image. The CNN-based backbone in YOLOv4 is replaced by the Swin Transformer in the proposed algorithm. The feature-fusing neck and predicting head of YOLOv4 is remained. The proposed model was trained and evaluated in the COCO dataset. Experiments show that our method can significantly improve the accuracy of object detection under special conditions. Equipped with our method, the object detection precision for cars and person is improved by 1.75%, and the detection precision for car and person reach 89.04% and 94.16%, respectively.

Suggested Citation

  • Kewei Lu & Fengkui Zhao & Xiaomei Xu & Yong Zhang, 2023. "An object detection algorithm combining self-attention and YOLOv4 in traffic scene," PLOS ONE, Public Library of Science, vol. 18(5), pages 1-18, May.
    • Handle: RePEc:plo:pone00:0285654
    DOI: 10.1371/journal.pone.0285654
    as

    Download full text from publisher

    File URL: https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0285654
    Download Restriction: no
    File URL: https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0285654&type=printable
    Download Restriction: no
    File URL: https://libkey.io/10.1371/journal.pone.0285654?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    References listed on IDEAS

    as
    1. Zhun Cheng & Zhixiong Lu, 2021. "Research on Load Disturbance Based Variable Speed PID Control and a Novel Denoising Method Based Effect Evaluation of HST for Agricultural Machinery," Agriculture, MDPI, vol. 11(10), pages 1-18, October.
    Full references (including those not matched with items on IDEAS)

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Zhenguo Zhang & Quanfeng Guo & Jin He & Minyi Zhao & Zhenyu Xing & Chao Zeng & Han Lin & Quanyu Wang, 2023. "Design and Experiment of Side-Shift Stubble Avoidance System for No-Till Wheat Seeder Based on Deviation-Perception Fusion Technology," Agriculture, MDPI, vol. 13(1), pages 1-18, January.
    2. Francesco Mocera & Aurelio Somà & Salvatore Martelli & Valerio Martini, 2023. "Trends and Future Perspective of Electrification in Agricultural Tractor-Implement Applications," Energies, MDPI, vol. 16(18), pages 1-36, September.
    3. Jie Tian & Mingfei Yang, 2023. "Hierarchical control of differential steering for four-in-wheel-motor electric vehicle," PLOS ONE, Public Library of Science, vol. 18(6), pages 1-20, June.
    4. Meng Yang & Xiaoxu Sun & Xiaoting Deng & Zhixiong Lu & Tao Wang, 2023. "Extrapolation of Tractor Traction Resistance Load Spectrum and Compilation of Loading Spectrum Based on Optimal Threshold Selection Using a Genetic Algorithm," Agriculture, MDPI, vol. 13(6), pages 1-20, May.

    More about this item

    Statistics

    Access and download statistics

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:plo:pone00:0285654. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: plosone (email available below). General contact details of provider: https://journals.plos.org/plosone/ .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.