基于多尺度压缩卷积神经网络模型的表面缺陷快速检测

来源：计算机集成制造系统杂志2022年第11期北京时间：

作者：廉家伟, 何军红, 牛云, 王天泽

单位：西北工业大学航海学院

摘要：应用基于各种卷积神经网络算法的图像处理技术对表面缺陷进行检测识别,不仅可以降低人工的成本,还可以大大提高效率和准确度。但是,当前比较热门的图像处理技术普遍存在计算量大、存储成本高以及模型复杂等特点,与工业应用所要求的高实时性以及有限的计算资源相悖。因此,提出一种基于多尺度压缩卷积神经网络模型(MC-CNN)进行表面缺陷快速检测,通过网络结构优化、知识蒸馏、网络修剪以及参数量化等压缩方法对网络进行多尺度压缩。实验表明,所提方法能够在保证卷积神经网络识别准确度的同时,大幅度提高模型的识别效率,缩小模型的体积,便于在工业现场各种实时性要求高而且存储和计算资源有限的场景中应用。

关键词：表面缺陷检测,卷积神经网络,多尺度,网络压缩

基金资助：国防科技工业强基工程军工智能制造专项行动计划资助项目。