买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
摘要:本发明公开了一种基于注意力机制和迁移学习的轮胎花纹分类方法,包括通过直方图均衡化,数据增强处理原始轮胎图片得到训练样本集,构建包括ResNet50网络、通道空间注意力机制、池化层和全连接层的训练模型,通过类别标签的交叉熵损失函数训练得到轮胎花纹分类模型。本发明还提供了对轮胎花纹分类模型进行压缩的方法,通过知识蒸馏方法,将轮胎花纹分类模型作为教师网络训练得到学生网络,通过交叉熵损失函数的训练的压缩后轮胎花纹分类模型,本发明还公开了一种基于注意力机制和迁移学习的轮胎花纹分类装置。
主权项:1.一种基于注意力机制和迁移学习的轮胎花纹分类方法,其特征在于,包括:1获得带有真实标签的原始轮胎图片集,对每个原始轮胎图片进行自适应直方图均衡化处理得到变换后轮胎图片集,基于变换后轮胎图片集采用数据增强方法得到衍生图片集,衍生图片集和变换后轮胎图片集构建训练样本集;2构建训练模型,包括ResNet50网络、通道空间注意力机制、池化层和全连接层,通过ImageNet数据集训练得到ResNet50网络,保留ResNet50网络中的特征提取层,ResNet50网络中的特征提取层用于提取训练样本的特征图,通道空间注意力机制,用于对输入的特征图中的轮胎纹理分类区域进行特征提取得到轮胎纹理分类特征图,将轮胎纹理分类特征图依次通过池化层和全连接层得到训练样本的多个预测类别标签概率;3通过多个预测类别标签概率和真实标签构建第一交叉熵损失函数,通过第一交叉熵损失函数训练并更新训练模型的参数得到轮胎花纹分类模型;4应用时,将轮胎图片输入至轮胎花纹分类模型得到轮胎图片的轮胎花纹类型。
全文数据:
权利要求:
百度查询: 浙江理工大学 一种基于注意力机制和迁移学习的轮胎花纹分类方法和装置
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。