摘要
磁瓦图像具有光照不均、表面纹理复杂、对比度低等特点。针对传统的缺陷检测算法难以准确分割其中缺陷的问题,提出基于掩膜区域卷积网络(Mask R-CNN)的缺陷检测算法。算法首先通过限制对比度的自适应直方图均衡化方法,对图像预处理。然后,采用残差网络50(ResNet50)构建特征金字塔网络(FPN)获取图像信息并提取特征,再采用区域建议网络(RPN)提取缺陷区域的感兴趣区域得到相应的锚框,并通过全卷积神经网络(FCN)来对感兴趣区域内部的像素类别进行预测以实现缺陷分割,最后通过网络的全连接层实现每个感兴趣区域所属类别和相应锚框坐标的预测。实验结果表明,该算法具有较强的泛化能力,可以对表面存在大量纹理复杂、光照不均、和低对比度的磁瓦图像进行精确的缺陷分割,具有较强的鲁棒性。
- 单位