|国家科技期刊平台
首页|期刊导航|郑州大学学报(工学版)|基于改进YOLOv8s算法的胸环靶弹孔检测技术

基于改进YOLOv8s算法的胸环靶弹孔检测技术OA北大核心CSTPCD

Bullet Hole Detection Technology of Chest Bitmap Based on Improved YOLOv8s Algorithm

中文摘要英文摘要

为了解决传统胸环靶弹孔检测技术在自然条件下易受光照强度、复杂背景影响的问题,设计了一种基于YOLOv8s的改进算法.首先,在数据集的制作过程中引入图形分割将背景与胸环靶分离,避免了复杂环境对弹孔识别精度的影响.其次,为提升模型对弹孔的检测能力,在 C2f中引入 CBAM 注意力机制,通过对空间和通道特征赋予不同的权值提高网络对弹孔目标的识别能力;增加检测尺度为 160×160 的小目标输出层,减少了弹孔特征在下采样过程中的信息损失并降低弹孔漏检的概率;考虑到原有卷积层对小目标不敏感,采用 SPD-Conv模块替换原有卷积层,提取更多的特征信息提升检测精度.最后,将边界框损失函数更改为 WIoU以减弱正负样本数量不均衡的影响,提高了预测框的回归精度.在自制胸环靶数据集的实验结果表明:改进算法的准确率 P 为 96.9%、召回率R为 96.4%、平均精度 mAP50 为 98.0%,相较于原算法,分别提升 8.8 百分点、25.4 百分点、15.3 百分点.实验结果证明改进的 YOLOv8s模型在复杂环境和密集弹孔的检测方面具有更好的性能.

Traditional chest bitmap bullet hole detection technology was easily affected by light intensity and com-plex background in natural conditions.In order to solve the proplem an improved algorithm based on YOLOv8s was designed in this study.Firstly,in order to avoid the impact of complex environment on the accuracy of bullet hole recognition,graph segmentation was introduced in the process of data set production to separate the background from the chest bitmap.Secondly,in order to improve the detection ability of the model to the bullet hole,CBAM attention mechanism was introduced into C2f,and the recognition ability of the network to the target bullet hole was improved by giving different weights to the spatial and channel characteristics.In order to reduce the information loss of bullet hole characteristics in the down sampling process and reduce the probability of missing bullet hole detection,the detection scale was increased to 160×160 small target output layer.Considering that the original con-volutional layer was not sensitive to small targets,the SPD-Conv module was used to replace the original convolu-tional layer to extract more feature information to improve detection accuracy.Finally,the loss function of the boun-ding box was changed to WIoU to weaken the influence of the unbalanced number of positive and negative samples and improve the regression accuracy of the prediction box.The experimental results on the self-made chest bitmap data set showed that the accuracy rate P of the improved algorithm was 96.9%,the recall rate R was 96.4%,and the average accuracy mAP50 was 98.0%,which were improved by 8.8 percentage points,25.4 percentage points,and 15.3 percentage points respectively,compared with the original algorithm.The experimental results showed that the improved YOLOv8s model had better performance in the detection of complex environment and dense bullet holes.

苏宇锋;边锋;张玉堂

郑州大学 机械与动力工程学院,河南 郑州 450001郑州纬达自动化科技有限公司,河南 郑州 450052

计算机与自动化

YOLOv8s弹孔检测CBAM注意力机制损失函数SPD-Conv

YOLOv8sbullet hole detectionCBAM attention mechanismloss functionSPD-Conv

《郑州大学学报(工学版)》 2024 (005)

16-22 / 7

国家重点研发计划项目(2018YFB0104100);河南省科技攻关项目(232102220005)

10.13705/j.issn.1671-6833.2024.05.013

评论