一种基于迁移学习的陶瓷瓶缺陷检测方法及系统

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种基于迁移学习的陶瓷瓶缺陷检测方法及系统
申请号:CN202410889875
申请日期:2024-07-04
公开号:CN118781073A
公开日期:2024-10-15
类型:发明专利
摘要
本发明公开了一种基于迁移学习的陶瓷瓶缺陷检测方法及系统,方法包括以下步骤:获取瓷砖缺陷图像集和陶瓷瓶缺陷图像集,并对瓷砖缺陷图像集和陶瓷瓶缺陷图像集进行处理;构建原始ResNet网络模型,并对原始ResNet网络模型进行改进,得到改进后ResNet网络模型;基于处理后的瓷砖缺陷图像集训练原始ResNet网络模型,基于处理后的陶瓷瓶缺陷图像集训练改进后ResNet网络模型;将训练好的原始ResNet网络模型通过迁移学习引入至训练好的改进后ResNet网络模型,得到陶瓷瓶缺陷检测模型;采集陶瓷瓶图像,并基于陶瓷瓶缺陷检测模型对陶瓷瓶缺陷进行检测。
技术关键词
ResNet网络 陶瓷瓶 缺陷检测方法 焦点损失函数 瓷砖 子模块 注意力机制 模型训练模块 图像处理模块 多尺度特征 通道 缺陷检测系统 全局平均池化 加权平均法 多层感知机 探头