A lightweight target detection method based on improved YOLOv3
By pruning and structural improvements to the YOLOv3 network model, including adjusting the residual structure and building branches, and using CIoU loss function and DIoU-NMS, the problem of difficult deployment of deep learning object detection algorithms in embedded systems is solved, and efficient and lightweight object detection is achieved, suitable for equipment with limited resources.
Patent Information
- Application Number
- CN202210287339.9
- Authority / Receiving Office
- CN · China
- Patent Type
- Patents(China)
- Current Assignee / Owner
- Filing Date
- 2022-03-22
- Publication Date
- 2025-03-28
- Estimated Expiration
- 2042-03-22
AI Technical Summary
Deep learning object detection algorithms are difficult to deploy in embedded systems, mainly due to the huge amount of model operations and the difficulty in running on devices with limited computing resources and storage resources, which makes it difficult to meet real-time requirements in actual application scenarios.
By pruning the backbone part of the YOLOv3 network model, the original residual structure is adjusted to a sparse residual structure, and a convolutional branch and pooled branch are constructed, the loss function is adjusted to a CIoU loss function, and DIoU-NMS is used in the post-processing module to reduce the model size and maintain detection accuracy.
It achieves the reduction of the model size to 3.17MB while ensuring detection accuracy and real-time performance, which is 98.7% lower than the YOLOv3 algorithm. It has high lightweight and practicality, and is suitable for the deployment of embedded devices.