关键词:
泄漏检测
实例分割
目标检测
自注意力机制
置信度传播簇机制
通道剪枝
摘要:
火电厂关键区域的管道、阀门等设备时常发生油液、蒸汽等物质的泄漏故障.为了提高火电厂管道及阀门泄漏故障的分割与检测精度,加快模型推理速度,提出一种基于改进YOLOv7的火电厂管道及阀门泄漏分割与检测算法,通过在YOLOv7网络中加入相关实例分割模块,实现实例分割与目标检测任务的并行;通过融入极化自注意力机制和可分离的视觉深度自注意力模块(separable vision transformer),弱化复杂背景的干扰,强化对泄漏区域的边缘提取;在后处理阶段运用置信度传播簇机制(confidence propagation cluster, CP-cluster),提高模型定位的准确性;在颈部网络使用幽灵卷积降低特征冗余,并通过通道剪枝技术压缩模型,实现模型轻量化.实验结果表明,在加入各项改进后,算法分割任务与检测任务的mAP@0.5:0.95指标分别达到75.7%、82.2%,相较于基础模型,指标分别提高了11.9%、7.1%,且模型参数量减少了30.3%,可有效地应用于电厂的实际生产环境中.