Back to Search Start Over

An improved method MSS-YOLOv5 for object detection with balancing speed-accuracy

Authors :
Yaping He
Yingying Su
Xiaofeng Wang
Jun Yu
Yu Luo
Source :
Frontiers in Physics, Vol 10 (2023)
Publication Year :
2023
Publisher :
Frontiers Media S.A., 2023.

Abstract

For deep learning-based object detection, we present a superior network named MSS-YOLOv5, which not only considers the reliability in complex scenes but also promotes its timeliness to better adapt to practical scenarios. First of all, multi-scale information is integrated into different feature dimensions to improve the distinction and robustness of features. The design of the detectors increases the variety of detection boxes to accommodate a wider range of detected objects. Secondly, the pooling method is upgraded to obtain more detailed information. At last, we add the Angle cost and assign new weights to different loss functions to accelerate the convergence and improve the accuracy of network detection. In our network, we explore four variants MSS-YOLOv5s, MSS-YOLOv5m, MSS-YOLOv5x, and MSS-YOLOv5l. Experimental results of MSS-Yolov5s show that our technique improves mAP on the PASCAL VOC2007 and PASCAL 2012 datasets by 2.4% and 2.9%, respectively. Meanwhile, it maintains a fast inference speed. At the same time, the other three models have different degrees of performance improvement in terms of balancing speed and precision in challenging detection regions.

Details

Language :
English
ISSN :
2296424X
Volume :
10
Database :
Directory of Open Access Journals
Journal :
Frontiers in Physics
Publication Type :
Academic Journal
Accession number :
edsdoj.9fe6a9701dad42f39bad3a5f852486db
Document Type :
article
Full Text :
https://doi.org/10.3389/fphy.2022.1101923