一种基于轻量化卷积神经网络的划痕检测方法及系统

摘要

本发明提出了一种基于轻量化卷积神经网络的划痕检测方法及系统。该方法为：采集样本，对样本进行预处理；搭建轻量化网络模型，所述轻量化网络包括由2至4个用于划痕特征识别的stage组成的骨干网络，每个stage的输出分别输入与每个stage一一对应的预测模块进行置信度初步预测，每个预测模块的输出连接一对初步预测的置信度进行融合的融合层；利用预处理后的样本于该轻量化网络模型中进行划痕检测。本方法去掉了VGG网络的最后一个阶段，减少了网络的参数量，提高了检测速度。