关键词:
电缆开关
X射线数字成像
YOLOv8n
空间与通道重建卷积
幻影卷积
注意力机制
摘要:
【目的】针对电缆开关图像检测算法存在模型复杂、检测效率低等问题,提出一种改进的基于全局注意力机制和幻影卷积的轻量级YOLOv8n模型——SGG-YOLOv8n,以实现X射线数字成像(x-ray digital radiography,X-DR)电缆开关故障的快速精准诊断。【方法】以YOLOv8n模型为基础,在其骨干网路和颈部网络引入空间与通道重建卷积(spatial and channel reconstruction convolution,SCConv),即将C2f模块的Bottleneck替换成SCConv模块,以增强模型感受野的灵活性,提升其检测效率和性能;同时,在YOLOv8n模型的颈部网络中,引入幻影卷积(ghost convolution,GhostConv)代替原有的Conv模块,以有效减少模型计算量和参数量,从而实现模型的轻量化;此外,在检测头前引入全局注意机制(global attention mechanism,GAM),进一步增强模型对全局特征信息的感知能力。【结果】在基于X-DR电缆开关图像数据集上的实验表明,与YOLOv8n相比,本文提出的SGG-YOLOv8n模型参数量降低了0.39 M,计算量下降了0.9 GFLOPs,精度提升了1.4%,召回率提升了1.4%,mAP@50提升了1.7%,mAP@50-95提高了1.2%,模型处理速度达65帧/秒。【结论】本文方法已实际应用于基于X-DR的电缆开关故障诊断中。模型应用后,电缆开关故障诊断准确率也从原来的87.9%提升至91.4%。