Back to Search Start Over

LightYOLO-S: a lightweight algorithm for detecting small targets.

Authors :
Zihan, Liu
xu, Wu
Linyun, Zhang
Panlin, Yu
Source :
Journal of Real-Time Image Processing; Jul2024, Vol. 21 Issue 4, p1-11, 11p
Publication Year :
2024

Abstract

Small target detection tasks are the focus and difficulty of target detection tasks. Methods to improve detection accuracy are often accompanied by drawbacks, such as a high number of parameters, computational effort, and latency. Therefore, this paper proposes the LightYOLO-S target detection algorithm based on YOLOv8s, which achieves high accuracy in small target detection tasks. First, the proposed LightC2f module reduces the overall number of parameters, computation, and inference time of the model while maintaining the same plug-and-play characteristics as the C2f module. Second, the proposed Wise-DIoU loss function, which speeds up model convergence and improves accuracy without increasing the number of parameters or computation. Third, the proposed Dynamic Sampler counts the IoU scores and classification scores of the screened training samples to adjust the sample allocation function, so that the model can obtain the best training samples during training.The results of experimental studies on the VisDrone2019 UAV aerial photography dataset and the DOTA remote sensing dataset indicate that LightYOLO-S exhibits greater accuracy and faster operation speed than the current state-of-the-art detection algorithms. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
18618200
Volume :
21
Issue :
4
Database :
Complementary Index
Journal :
Journal of Real-Time Image Processing
Publication Type :
Academic Journal
Accession number :
177917671
Full Text :
https://doi.org/10.1007/s11554-024-01485-x