来源:市场资讯

(来源:计算机视觉研究院 OpenCV与AI深度学习)

https://pmc.ncbi.nlm.nih.gov/articles/PMC13259083/pdf/sensors-26-03418.pdf

计算机视觉研究院专栏

Column of Computer Vision Institute

本文提出轻量化检测模型YOLO-LSS,针对低光复杂的管道场景量身优化,仅 7.05MB 体积就能跑出 188FPS 的实时速度,mAP50 提升至 79.05%,为管道缺陷智能检测提供了高精度、易部署的全新方案。

PART/1

痛点

金属管道长期埋于地下,内部环境封闭昏暗,检测面临三重核心难题:

:传统 CCTV 依赖专业人员逐帧判读视频,人力成本高、检测周期长,结果受主观经验影响大;

:管道内低光照、高噪声、管壁反光、背景杂乱,普通视觉算法很难精准提取缺陷特征;

:管道巡检机器人多为边缘设备,算力和存储空间有限,大模型难以落地,小模型又存在精度不足、小目标漏检多的问题。

PART/2

创新

核心升级:三大改造,打造管道专属检测模型

研究团队以轻量化的 YOLOv8n 为基础框架,从特征提纯、多尺度感知、难例优化三个维度全面改造,提出 YOLO-LSS(YOLOv8n-LskBlock-SlideLoss-SCSA)模型。

【YOLO-LSS 模型整体架构图】
打开网易新闻 查看精彩图片
【YOLO-LSS 模型整体架构图】

1. SCSA 协同注意力:给模型装上 “降噪放大镜”

针对管道图像噪声多、背景杂的问题,模型引入 SCSA 空间 - 通道协同注意力机制:

  • 空间维度通过多尺度深度 1D 卷积,提取不同层级的空间特征,过滤无效背景干扰;

  • 通道维度采用渐进式自注意力,强化缺陷相关的特征通道,抑制冗余信息。双维度协同作用,相当于先给图像 “降噪提纯”,再精准聚焦缺陷区域,大幅提升复杂背景下的特征提取能力。

2. LskBlock 大核选择模块:多尺度缺陷 “一网打尽”

管道缺陷尺度差异极大,从针尖大小的腐蚀点到延伸数米的裂缝都有。团队在 Neck 网络的 P3、P4、P5 三个特征层级融入 LskBlock 模块:通过分解式大核深度卷积扩大感受野,同时支持空间自适应选择,可根据目标大小动态调整感知范围 —— 小缺陷抓细节纹理,大缺陷看全局轮廓,完美适配管道内多尺度缺陷的检测需求。

3. SlideLoss 滑动损失:专攻 “疑难样本”

管道数据中,大量简单背景样本会 “淹没” 困难缺陷的梯度,导致模型学不到关键特征。团队用 SlideLoss 函数优化分类损失:基于预测置信度与 IoU 实现分段自适应加权,给决策边界附近的困难样本(模糊缺陷、遮挡目标、低对比度目标)加大损失权重,给简单的背景样本降低权重,引导模型重点攻克难分场景,显著减少误检和漏检。

PART/3

实验

实验实测:精度速度双在线,轻量易部署

团队基于 Roboflow 公开管道数据集开展实验,样本包含腐蚀、沉积、氧化物脱落、穿孔四类缺陷,所有图像均为低光环境采集,自带高噪声、低对比度特性,高度贴合真实巡检场景。

1. 整体性能:主流模型对比全面领先

打开网易新闻 查看精彩图片

【不同模型测试集性能对比表】和 YOLO 全系列、SSD、Faster R-CNN、RT-DETR 等主流算法相比,YOLO-LSS 表现亮眼:

  • 精度领先

    :mAP50 达 79.05%,mAP50:95 达 53.7%,相比原生 YOLOv8n 分别提升 2.86% 和 7.19%,精准度超过多数同量级模型;

  • 速度拉满

    :推理速度达 188FPS,单张图像仅需 5.32 毫秒,远超工业实时检测 30FPS 的标准;

  • 极致轻量

    :模型体积仅 7.05MB,比 RT-DETR 小近 18 倍,算力需求极低,可直接部署在管道巡检机器人等边缘终端。

2. 可视化效果:漏检更少,置信度更高

【YOLOv8n 与 YOLO-LSS 检测效果对比图】
打开网易新闻 查看精彩图片
【YOLOv8n 与 YOLO-LSS 检测效果对比图】

从实测结果可见,原生 YOLOv8n 容易漏检的细小缺陷,YOLO-LSS 均能精准识别,且检测置信度明显提升。热力图可视化也证实,改进后模型的注意力更集中在缺陷核心区域,受背景噪声和管壁反光的干扰显著降低。

3. 消融实验:每个模块都发挥关键作用

【模型消融实验结果表】
打开网易新闻 查看精彩图片
【模型消融实验结果表】

通过逐一验证三个模块的贡献,证实 SCSA、LskBlock、SlideLoss 均可独立提升检测效果,三者组合后形成 “特征降噪 - 尺度适配 - 难例优化” 的完整闭环,实现了性能最大化。

PART/4

总结与展望

这项研究提出的 YOLO-LSS 模型,以极小的算力代价实现了复杂管道环境下缺陷检测的精度跃升,为地下管道智能巡检提供了轻量化、高实时性的算法方案,可直接落地于管道检测机器人、在线监测系统等场景。

目前模型在氧化物脱落等极小缺陷上仍有提升空间,团队表示,未来将引入超分辨率、特征金字塔增强等技术,进一步优化微缺陷的检测能力;同时将拓展多场景数据集,提升模型在不同材质、不同环境管道中的泛化性。