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.

An Improved Target Detection General Framework Based on Yolov4

Target detection for remote sensing based on the enhanced YOLOv4 with improved BiFPN

YOLOv4-5D: An Effective and Efficient Object Detector for Autonomous Driving

Research on Mask Detection Method Based on Yolov8

An Improved Mobilenetv3-Yolov5 Infrared Target Detection Algorithm Based on Attention Distillation

An Improved YOLO for Road and Vehicle Target Detection Model

4AC-YOLOv5: an improved algorithm for small target face detection

An Aerial Image Detection Algorithm Based on Improved YOLOv5

YOLOV4_CSPBi: Enhanced Land Target Detection Model

Improved YOLO Network for Free-Angle Remote Sensing Target Detection

Target Detection of Remote Sensing Image Based on an Improved YOLOv5

High-precision real-time autonomous driving target detection based on YOLOv8

Large Span Sizes and Irregular Shapes Target Detection Methods Using Variable Convolution-Improved YOLOv8

Gated Channel Attention Mechanism YOLOv3 Network for Small Target Detection

An improved lightweight object detection algorithm for YOLOv5

Lightweight target detection algorithm based on YOLOv4

Real-time object detection method based on YOLOv5 and efficient mobile network

YOLOv8-CAS: An improvement of multi-class target detection algorithm based on YOLO

Improvement of Remote sensing image target detection algorithm based on YOLO V5

YOLO-APDM: Improved YOLOv8 for Road Target Detection in Infrared Images