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.

Lightweight Road Vehicle Detection Algorithm Based on Improved YOLOv8

Lightweight Vehicle Detection Based on Improved YOLOv5s

A Lightweight Vehicle Detection Method Fusing GSConv and Coordinate Attention Mechanism

A lightweight vehicles detection network model based on YOLOv5

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

Application Research on Lightweight Vehicle Detection Based on YOLO

Fast vehicle detection algorithm based on lightweight YOLO7-tiny

Improved YOLOv8-based Vehicle Detection Method for Road Monitoring and Surveillance

An Improved YOLO for Road and Vehicle Target Detection Model

A Lightweight Vehicle-Pedestrian Detection Algorithm Based on YOLOv5s

Light “You Only Look Once”: An Improved Lightweight Vehicle-Detection Model for Intelligent Vehicles under Dark Conditions

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

YOLOv8-FDD: A Real-time Vehicle Detection Method Based on Improved YOLOv8

Research on vehicle detection based on improved YOLOX_S.

Vehicle and Pedestrian Detection Method Based on Improved YOLOv4-tiny

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

Road Traffic Small Target Detection Algorithm Based on Lightweight YOLO V5

Vehicle–Pedestrian Detection Method Based on Improved YOLOv8

YOLOv8-QSD: An Improved Small Object Detection Algorithm for Autonomous Vehicles Based on YOLOv8

Research on vehicle detection based on improved YOLOv8 network

An improved lightweight object detection algorithm for YOLOv5