摘要
本发明公开了一种基于迁移学习的陶瓷瓶缺陷检测方法及系统,方法包括以下步骤:获取瓷砖缺陷图像集和陶瓷瓶缺陷图像集,并对瓷砖缺陷图像集和陶瓷瓶缺陷图像集进行处理;构建原始ResNet网络模型,并对原始ResNet网络模型进行改进,得到改进后ResNet网络模型;基于处理后的瓷砖缺陷图像集训练原始ResNet网络模型,基于处理后的陶瓷瓶缺陷图像集训练改进后ResNet网络模型;将训练好的原始ResNet网络模型通过迁移学习引入至训练好的改进后ResNet网络模型,得到陶瓷瓶缺陷检测模型;采集陶瓷瓶图像,并基于陶瓷瓶缺陷检测模型对陶瓷瓶缺陷进行检测。
技术关键词
ResNet网络
陶瓷瓶
缺陷检测方法
焦点损失函数
瓷砖
子模块
注意力机制
模型训练模块
图像处理模块
多尺度特征
通道
缺陷检测系统
全局平均池化
加权平均法
多层感知机
探头