Abstract:Currently, with the widespread application of embedded technology and the continuous improvement of computational power in mobile terminals, the efficient deployment of algorithms on embedded devices, while maintaining high accuracy and minimizing model size, has become a research hotspot. This paper addresses the challenges of deploying the YOLOv8 algorithm on embedded devices and proposes a novel lightweight object detection algorithm focusing on small object detection. We optimize the model through two key strategies, aiming to achieve lightweight deployment and improve the accuracy of small object detection. Firstly, GhostNet is introduced as the backbone network for YOLOv8 in order to achieve lightweight deployment. By using some cost effective operations to generate redundant feature maps, we not only reduce the number of model parameters while ensuring better detection results, but also improve the speed of the model. Secondly, a new multi-scale attention module is designed to enhance the network's acquisition of crucial information for small targets, which includes a multi-scale fusion attention mechanism and the Soft-NMS algorithm. The multi-scale fusion attention mechanism captures key features of discriminative small targets in the feature map tensor from both spatial and channel dimensions, suppressing non-key information, reducing the impact of complex and unimportant information in the image, enhancing the network model's learning ability for important features of small targets. The Soft-NMS method improves accuracy by significantly reduces false positives in the detection results. To validate the performance of our proposed method, we conducted validation experiments on the PASCAL VOC dataset and evaluated the model's generalization ability on the MS COCO dataset. The experiments results demonstrate that our model achieves a significant improvement in small object detection, with a 5.41% increase in detection accuracy compared to the existing YOLOv8. Meanwhile, FLOPs are reduced by 49.62%, and the number of model parameters is reduced by 48.66%. These results fully confirm the effectiveness of our innovative method in achieving both lightweight deployment and significant efficacy in small object detection tasks.

Embedded Object Detection System Based on Deep Neural Network.

A Lightweight SE-YOLOv3 Network for Multi-Scale Object Detection in Remote Sensing Imagery.

Mobile Centernet for Embedded Deep Learning Object Detection

Multiclass objects detection algorithm using DarkNet-53 and DenseNet for intelligent vehicles

Fast and Accurate Object Detection in Remote Sensing Images Based on Lightweight Deep Neural Network

Developing a Compressed Object Detection Model based on YOLOv4 for Deployment on Embedded GPU Platform of Autonomous System

Tiny SSD: A Tiny Single-Shot Detection Deep Convolutional Neural Network for Real-Time Embedded Object Detection

Towards Real-Time Object Detection on Embedded Systems.

Efficient Object Detection Based on Masking Semantic Segmentation Region for Lightweight Embedded Processors

Embedded YOLO: A Real-Time Object Detector for Small Intelligent Trajectory Cars

Real-Time Object Detection Using Pre-Trained Deep Learning Models MobileNet-SSD

Mob-YOLO: A Lightweight UAV Object Detection Method

An Efficient Model Compression Method for CNN Based Object Detection

Object Detection Implementation and Optimization on Embedded GPU System.

MobileYOLO: Real-Time Object Detection Algorithm in Autonomous Driving Scenarios

LAYN: Lightweight Multi-Scale Attention YOLOv8 Network for Small Object Detection

Efficient Object Detection Using Semantic Region of Interest Generation with Light-Weighted LiDAR Clustering in Embedded Processors

A Lightweight Object Detection Network for Real-Time Detection of Driver Handheld Call on Embedded Devices

Multi-scale Lightweight Neural Network for Real-Time Object Detection.

Research on the Construction of an Efficient and Lightweight Online Detection Method for Tiny Surface Defects through Model Compression and Knowledge Distillation

High-resolution processing and sigmoid fusion modules for efficient detection of small objects in an embedded system