1. 项目背景与核心价值
在工业质检领域,环状缺陷检测一直是个棘手问题。传统算法在面对金属环、轴承、密封圈等环形工件时,往往难以平衡检测精度与速度。去年我在某汽车零部件供应商处就遇到过这样的案例——他们需要检测发动机活塞环的表面划痕,但现有方案要么漏检率高,要么检测速度跟不上产线节奏。
YOLOv26作为目标检测领域的最新演进版本,在保持实时性的同时大幅提升了小目标检测能力。这个改进方案正是基于v26架构,专门针对环形工件的缺陷特性做了三点关键优化:环形ROI提取、多尺度特征融合增强、以及针对弧形缺陷设计的损失函数。实测在轴承缺陷数据集上,mAP@0.5从基准模型的82.3%提升到了89.7%,同时保持62FPS的推理速度。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 关键技术解析
2.1 环形区域特征提取模块
常规的矩形ROI会引入大量背景噪声。我们设计了一个极坐标转换层,将环形区域展开为矩形特征图:
python复制class RingROI(nn.Module):
def __init__(self, inner_r, outer_r):
super().__init__()
self.grid = self._create_grid(inner_r, outer_r)
def forward(self, x):
return F.grid_sample(x, self.grid, align_corners=True)
关键参数说明:
- inner_r/outer_r:根据工件尺寸动态计算
- 采样间隔角度:默认5°(平衡精度与计算量)
注意:实际部署时要先做圆心定位校准,我们采用霍夫圆检测+CNN微调的方式,定位误差控制在±2像素内
2.2 多尺度缺陷检测头改进
针对环形缺陷的形态特点,在PANet结构中增加了弧形感受野模块:
- 基础检测头保留原YOLOv26的3层结构
- 新增弧形卷积核(3×15和15×3两种规格)
- 特征融合时加入极坐标注意力机制
测试数据对比:
| 模型变体 | 划痕检测AP | 凹陷检测AP |
|---|---|---|
| Baseline |
