Small Target Vehicle Detection Algorithm Based on Improved YOLOv5s

In recent years, YOLOv5s has achieved notable success in object detection. However, its performance in detecting small objects still requires improvement. This paper introduces three key enhancements aimed at improving YOLOv5s for small object detection. Firstly, a detection head for small objects is added, transitioning from three-scale to four-scale detection, thereby enhancing the model's ability to detect small-sized objects. Secondly, the CSP (Cross Stage Partial) structure is replaced by the C3GC (Cross Stage Partial with Global Context) structure, incorporating global context to boost feature representation and improve small object recognition. Finally, the loss function is optimized to Focal-EIOU (Extended Intersection over Union), better assessing bounding box overlap and improving localization accuracy. Experimental results demonstrate that the improved YOLOv5s model significantly outperforms its predecessor on public datasets, with marked improvements in precision and recall, especially in small object detection. These advancements provide an effective solution for small object detection and valuable insights for future research.

Authors

Institutions

Publication Details

Published
2024-10-25
DOI
https://doi.org/10.1109/cvci63518.2024.10830200
Primary Topic
Video Surveillance and Tracking Methods
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Small Target Vehicle Detection Algorithm Based on Improved YOLOv5s

Xincheng Sun, Xiaopeng Yang, Huifang Kong, Yibo Meng
Video Surveillance and Tracking Methods
article

Small Target Vehicle Detection Algorithm Based on Improved YOLOv5s

Xincheng Sun, Xiaopeng Yang, Huifang Kong, Yibo Meng
article en

Abstract

In recent years, YOLOv5s has achieved notable success in object detection. However, its performance in detecting small objects still requires improvement. This paper introduces three key enhancements aimed at improving YOLOv5s for small object detection. Firstly, a detection head for small objects is added, transitioning from three-scale to four-scale detection, thereby enhancing the model's ability to detect small-sized objects. Secondly, the CSP (Cross Stage Partial) structure is replaced by the C3GC (Cross Stage Partial with Global Context) structure, incorporating global context to boost feature representation and improve small object recognition. Finally, the loss function is optimized to Focal-EIOU (Extended Intersection over Union), better assessing bounding box overlap and improving localization accuracy. Experimental results demonstrate that the improved YOLOv5s model significantly outperforms its predecessor on public datasets, with marked improvements in precision and recall, especially in small object detection. These advancements provide an effective solution for small object detection and valuable insights for future research.

Hefei University of Technology (CN)
Openalex Percentile: Top 74%
Video Surveillance and Tracking Methods
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

Small Target Vehicle Detection Algorithm Based on Improved YOLOv5s — Xincheng Sun, Xiaopeng Yang, et al. · (2024) | TGRS Research Map | TGRS