改进FPN的Mask R-CNN工业表面缺陷检测

• 作者: 王海云 王剑平 张果 欧阳鑫 罗付华
• 来源期刊: 制造业自动化
• 学科: 工学
• 关键词: 表面缺陷检测 深度学习 Mask R-CNN 多尺度特征 准确定位
• 年，卷（期）: 2020,（12）
• 所属期刊栏目: 检测与监控
• 页码范围: 35-40,97
• 页数: 7页
• 分类号: TP391
• 语种: 中文

摘要

针对工业表面缺陷呈现形式存在多样性和随机性而导致难以准确检测的问题,提出改进FPN的Mask R-CNN表面缺陷检测算法.该算法使用ResNet-101提取缺陷图像各阶段的特征图,改进的FPN增加自底向上的反向连接路径对各阶段特征图进行充分融合,得到具有更强语义信息和定位信息的多尺度特征图;多尺度特征图输入RPN产生位置修正参数,进而得到候选区域;RolAlign网络基于多尺度特征图和候选区域映射产生固定尺度的候选区域特征图;候选区域特征图输入Faste R-CNN进行分类和回归,同时输入Mask分割网络进行掩膜预测.采用DAGM公开缺陷数据集验证本文算法性能,并与主流的检测算法进行对比实验.实验结果表明,所提算法与Faster R-CNN和Mask R-CNN相比,具有更优的检测性能,能更加准确地定位出各类缺陷,mAP可达到98.25％,在工业表面缺陷智能检测领域中具有较好的应用价值.