期刊
MACHINES
卷 11, 期 2, 页码 -出版社
MDPI
DOI: 10.3390/machines11020257
关键词
aerial images; bird nest detection; YOLOv5s; model deployment
This paper improves the lightweight model YOLOv5s to enhance the accuracy of bird nest model detection on transmission towers in aerial images. The OSA module and CBAM are utilized to enhance small target recognition, while reducing the parameters of the ASPP module to improve detection in complex backgrounds. The experiments demonstrate a significant increase in accuracy and only a slight decrease in speed of the improved YOLOv5s model.
To further improve the accuracy of bird nest model detection on transmission towers in aerial images without significantly increasing the model size and to make detection more suitable for edge-end applications, the lightweight model YOLOv5s is improved in this paper. First, the original backbone network is reconfigured using the OSA (One-Shot Aggregation) module in the VOVNet and the CBAM (Convolution Block Attention Module) is embedded into the feature extraction network, which improves the accuracy of the model for small target recognition. Then, the atrous rates and the number of atrous convolutions of the ASPP (Atrous Spatial Pyramid Pooling) module are reduced to effectively decrease the parameters of the ASPP. The ASPP is then embedded into the feature fusion network to enhance the detection of the targets in complex backgrounds, improving the model accuracy. The experiments show that the mAP (mean-Average Precision) of the fusion-improved YOLOv5s model improves from 91.84% to 95.18%, with only a 27.4% increase in model size. Finally, the improved YOLOv5s model is deployed into the Jeston Xavier NX, resulting in a model that runs well and has a substantial increase in accuracy and a speed of 10.2 FPS, which is only 0.7 FPS slower than the original YOLOv5s model.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据