Small target detection algorithm based on attention mechanism and data augmentation
Jiuxin Wang,Man Liu,Yaoheng Su,Jiahui Yao,Yurong Du,Minghu Zhao,Dingze Lu
DOI: https://doi.org/10.1007/s11760-024-03046-y
2024-02-26
Abstract:The detection of masks is of great significance to the prevention of occupational diseases such as infectious diseases and dust diseases. For the problems of small target size, large number of targets, and mutual occlusion in mask-wearing detection, a mask-wearing detection algorithm based on improved YOLOv5s is proposed in this paper. First, the ultralightweight attention mechanism module ECA is embedded in the neck layer to improve the accuracy of the model. Second, the influence of different loss functions (GIoU, CIoU, and DIoU) on the improved model is explored, and CIoU is determined as the loss function of the improved model. Besides, the improved model adopted the label smoothing method, which effectively improved the generalization ability of the model and reduced the risk of overfitting. Finally, the influence of data augmentation methods (Mosaic and Mixup) on model performance is discussed, and the optimal weight of data augmentation is determined. The proposed model is tested on the verification set, and the mean average precision (mAP), precision, and recall are 92.1%, 90.3%, and 87.4%, respectively. The mAP of the improved algorithm is 4.4% higher than that of the original algorithm.
engineering, electrical & electronic,imaging science & photographic technology