Location via proxy:   [ UP ]  
[Report a bug]   [Manage cookies]                
skip to main content
10.1145/3638264.3638276acmotherconferencesArticle/Chapter ViewAbstractPublication PagesmicmlConference Proceedingsconference-collections
research-article

Improved YoloV5 Model Target Detection Algorithm Based on Temporal Neural Networks

Published: 29 January 2024 Publication History

Abstract

In order to further improve the accuracy of object detection on datasets with time characteristics such as video and live streaming, this paper proposes an object detection algorithm that combines temporal neural networks. In this neural network, Mosaic data augmentation was used at the input end, and then entered the convolution section to obtain sufficient feature maps. These feature maps were continuously convolutionally processed to obtain feature maps of different scales for feature judgment. Features of different dimensions were normalized and sent to the temporal section. In the temporal part, the forgetting gate method was used to output the hidden layer information at the current time, which is a new feature map with preliminary judgment. Separate feature maps through 1 × 1 convolution expands the number of channels to obtain anchors with classification and detection information, and filters these anchors to screen positive and negative samples. After we generate prediction boxes for the target, we often generate a large number of redundant boundary boxes. Therefore, we need to remove boundary boxes with low positional accuracy and retain boundary boxes with high positional accuracy. Obtain the final prediction result. The model proposed in this article was compared with the YOLO and SSD models on the VisDrone dataset, and the improved algorithm achieved good accuracy improvement compared to the YOLO model and higher efficiency compared to the SSD model. The method proposed in this article provides a new approach and strategy for improving the integrity, accuracy, and universality of object detection tasks with temporal data.

References

[1]
Xingkui Zhu, . TPH-YOLOv5: Improved YOLOv5 Based on Transformer Prediction Head for Object Detection on Drone-captured Scenarios [J]. IEEE/CVF, 2021.
[2]
Redmon J,Divvala K S,Girshick B R, You Only Look Once: Unified, Real-Time Object Detection.[J]. CoRR,2015,abs/1506.02640.
[3]
Zijian Lin, Qunxu Lin, Gong Zhang, Garbage detection algorithm based on improved YOLO v5s [J]. Mechanical Engineer, 2023 (08): 54-57+60Jon M. Kleinberg. 1999. Authoritative sources in a hyperlinked environment. J. ACM 46, 5 (September 1999), 604–632. https://doi.org/10.1145/324133.324140
[4]
Hongwei Fan, Jinpeng Liu, Xiangang Cao, Intelligent multi-scale target detection method for early damage of coal, foreign objects, and conveyor belts under low illumination dust mist [J/OL]. Journal of Coal Industry :1-12[2023-08-13].
[5]
Minghao Cheng. Research on Target Detection in Gas Station Monitoring Scenarios Based on Improved YOLO Algorithm [D]. Northeast Petroleum University, 2023
[6]
Yu Yichun, Li Mingxu. autonomous vehicle target detection by improving YOOv5s [J/OL]. Computer system application: 1-9[2023-08-13].
[7]
Zhiyang Y,Chengdong L,Min Z, CCA-FPN: Channel and content adaptive object detection[J]. Journal of Visual Communication and Image Representation,2023,95.
[8]
Sumin H,Woo J J. Automatic Examination of Condition of Used Books with YOLO-Based Object Detection Framework[J]. 1 Department of Industrial Engineering, Seoul National University of Science and Technology, Seoul, 01811, Korea;2 Department of Data Science, Seoul National University of Science and Technology, Seoul, 01811, Korea,2023,47(2).
[9]
Dong X,Tong F X,Yang G, A detection method of spangle defects on zinc-coated steel surfaces based on improved YOLO-v5[J]. The International Journal of Advanced Manufacturing Technology,2023,128(1-2).
[10]
Mengzi H,Ziyang L,Jiong Y, Efficient-Lightweight YOLO: Improving Small Object Detection in YOLO for Aerial Images.[J]. Sensors (Basel, Switzerland),2023,23(14).
[11]
Wojciech L,Adam Ś,Tomasz K, YOLOv5 Drone Detection Using Multimodal Data Registered by the Vicon System.[J]. Sensors (Basel, Switzerland),2023,23(14).
[12]
Dongsheng C,Zexiang G,Olusola B, Multi-objective tracking for smart substation onsite surveillance based on YOLO Approach and AKCF[J]. Energy Reports,2023,9(S10).
[13]
Hui Guo, Haiyang Chen, Guomin Gao, Method for object detection and spatial positioning of safflower corolla based on YOLO v5m [J]. Journal of Agricultural Machinery,2023,54(07):272-281.
[14]
Chunjiang Zhao, Xuewen Liang, Helong Yu, An improved YOLO v7 based automatic identification and counting method for cage chickens/eggs [J]. Journal of Agricultural Machinery, 2023,54 (07): 300-312
[15]
Wu D,Liao M,Zhang W, YOLOP: You Only Look Once for Panoptic Driving Perception[J].Machine Intelligence Research,2022,19(06):550-562.

Index Terms

  1. Improved YoloV5 Model Target Detection Algorithm Based on Temporal Neural Networks

      Recommendations

      Comments

      Information & Contributors

      Information

      Published In

      cover image ACM Other conferences
      MICML '23: Proceedings of the 2023 International Conference on Mathematics, Intelligent Computing and Machine Learning
      December 2023
      109 pages
      ISBN:9798400709258
      DOI:10.1145/3638264
      Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than the author(s) must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected].

      Publisher

      Association for Computing Machinery

      New York, NY, United States

      Publication History

      Published: 29 January 2024

      Permissions

      Request permissions for this article.

      Check for updates

      Author Tags

      1. SSD
      2. Yolo
      3. target detection
      4. temporality

      Qualifiers

      • Research-article
      • Research
      • Refereed limited

      Conference

      MICML 2023

      Contributors

      Other Metrics

      Bibliometrics & Citations

      Bibliometrics

      Article Metrics

      • 0
        Total Citations
      • 13
        Total Downloads
      • Downloads (Last 12 months)13
      • Downloads (Last 6 weeks)2
      Reflects downloads up to 15 Oct 2024

      Other Metrics

      Citations

      View Options

      Get Access

      Login options

      View options

      PDF

      View or Download as a PDF file.

      PDF

      eReader

      View online with eReader.

      eReader

      HTML Format

      View this article in HTML Format.

      HTML Format

      Media

      Figures

      Other

      Tables

      Share

      Share

      Share this Publication link

      Share on social media