Abstract:Object detection based on deep learning has made great progress in the past decade and has been widely used in various fields of daily life. Model lightweighting is the core of deploying target detection models on mobile or edge devices. Lightweight models have fewer parameters and lower computational costs, but are often accompanied by lower detection accuracy. Based on YOLOv5s, this article proposes an improved lightweight target detection model, which can achieve higher detection accuracy with smaller parameters. Firstly, utilizing the lightweight feature of the Ghost module, we integrated it into the C3 structure and replaced some of the C3 modules after the upsample layer on the neck network, thereby reducing the number of model parameters and expediting the model’s inference process. Secondly, the coordinate attention (CA) mechanism was added to the neck to enhance the model’s ability to pay attention to relevant information and improved detection accuracy. Finally, a more efficient Simplified Spatial Pyramid Pooling—Fast (SimSPPF) module was designed to enhance the stability of the model and shorten the training time of the model. In order to verify the effectiveness of the improved model, experiments were conducted using three datasets with different features. Experimental results show that the number of parameters of our model is significantly reduced by 28% compared with the original model, and mean average precision (mAP) is increased by 3.1%, 1.1% and 1.8% respectively. The model also performs better in terms of accuracy compared to existing lightweight state-of-the-art models. On three datasets with different features, mAP of the proposed model achieved 87.2%, 77.8% and 92.3%, which is better than YOLOv7tiny (81.4%, 77.7%, 90.3%), YOLOv8n (84.7%, 77.7%, 90.6%) and other advanced models. When achieving the decreased number of parameters, the improved model can successfully increase mAP, providing great reference for deploying the model on mobile or edge devices.

Application Research on Lightweight Vehicle Detection Based on YOLO

A Lightweight Vehicle Detection Method Fusing GSConv and Coordinate Attention Mechanism

Lightweight Vehicle Detection Based on Improved YOLOv5s

A lightweight vehicles detection network model based on YOLOv5

Road Traffic Small Target Detection Algorithm Based on Lightweight YOLO V5

A Lightweight Vehicle-Pedestrian Detection Algorithm Based on YOLOv5s

Object Detection for Intelligent Driving Based on Improved YOLOv5

Lightweight Road Vehicle Detection Algorithm Based on Improved YOLOv8

Design of Real-Time Vehicle Detection Based on YOLOv4

Exploration of Vehicle Target Detection Method Based on Lightweight YOLOv5 Fusion Background Modeling

An improved lightweight object detection algorithm for YOLOv5

Lightweight Object Detection Based on Autonomous Driving

Lightweight object detection algorithm for robots with improved YOLOv5

Real-time Detection Algorithm for Non-Motorized Vehicles Based on D-YOLO Model.

Light-weighted Vehicle Detection Network Based on Improved YOLOv3-tiny

Fast vehicle detection algorithm based on lightweight YOLO7-tiny

Road Traffic Vehicle Detection Method Using Lightweight YOLOv5 and Attention Mechanism

A Lightweight Bus Passenger Detection Model Based on YOLOv5

YOLOdrive: A Lightweight Autonomous Driving Single-Stage Target Detection Approach

Lightweight Detection Network of Trunk Contents

A Lightweight Vehicle-Pedestrian Detection Algorithm Based on Attention Mechanism in Traffic Scenarios