计算机应用

作者：张京爱, 王江涛

单位：1. 淮北师范大学 物理与电子信息学院, 安徽 淮北 235000;2. 淮北师范大学 信息学院, 安徽 淮北 235000

关键词：磁瓦,表面缺陷检测,卷积神经网络,mixup,ResNet18

基金：安徽省高校自然科学研究项目（KJ2018ZD038，KJ2019B15）；安徽省质量工程项目（2019jxtd142）。

铁氧体磁瓦由于形状的不规则性和表面缺陷的多样性给基于计算机视觉的表面质量识别带来很大的挑战。针对该问题，将深度学习技术引入到磁瓦表面质量识别中，提出一种基于卷积神经网络的磁瓦表面质量识别系统。首先将磁瓦目标从采集到的图像中分割出来并进行旋转从而得到标准图像，然后把改进后的多尺度ResNet18作为骨干网络来设计识别系统。训练时，设计一种新颖的类内mixup操作来提高系统对样本的泛化能力。为了更加贴近实际应用场景，在考虑到光线变化、姿态差异等因素的前提下构建了磁瓦缺陷数据集。在自建的数据集中进行实验的结果表明，该系统可以达到97.9%的识别准确率，为磁瓦缺陷的自动识别提供了可行的思路。

来源：2021年第1期

《计算机应用》期刊编辑部

查看计算机应用杂志2021年第1期