买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:北京工业大学
摘要:该发明提出一种基于对比动态特征增量学习的废旧家电识别方法,针对废旧家电识别模型受到不同类别干扰,导致识别结果不稳定的问题。首先利用Resnet‑18构建废旧家电识别模型,计算得到废旧家电图像的深度特征,其次利用动态原型提供废旧家电的代表性特征,降低识别过程中类别的交叉干扰,最后利用废旧家电不同类别对比动态特征损失,结合动态原型的家电代表性特征,提升了识别精度,这种基于对比动态特征增量学习的废旧家电识别方法在实际废旧家电回收过程中,可以解决废旧家电由于数据量实时更新,从而导致识别结果不稳定的问题,实现了高准确性的废旧家电分类,为废旧家电回收行业提供技术支持。
主权项:1.一种基于对比动态特征增量学习的废旧家电识别方法,其特征在于,包括以下步骤:1采集及预处理废旧家电数据获取废旧家电图像数据集,共10类废旧家电图像:包括电路板、电线、铝、压缩机、冷凝器、铜管、外电机、阻燃、废杂物和内机扇,共S张JPG格式的图像,该数据集为废旧家电拆解部件图像,将废旧家电图像数据集中每张图像的像素归一化至[0,1];将归一化后的废旧家电图像数据集分为训练集和测试集,训练集为Xt=[X1,…,X5],t为任务数量,t=1,2,…,5,Xt图像总数量为S的70%,测试集为图像总数量为S的30%,当t=1时,进入初始训练任务,当t=2时,进入增量训练任务,每个任务选择2类废旧家电图像数据训练,中包含的废旧家电图像类别与Xt中的类别一致;2构建废旧家电分类网络模型构建基于ResNet-18的废旧家电分类网络:该网络包括五个部分,第一部分conv1由一个卷积核为3×3的卷积层,一个批量归一化处理层,一个ReLU激活函数层组成,通道数为64,第二部分conv2,通道数为64,第三部分conv3,通道数为128,第四部分conv4,通道数为256,第五部分conv5,通道数为512,均由两个残差模块组成,每个残差模块由一个卷积核为3×3的卷积层,一个ReLU激活函数层,一个卷积核为3×3的卷积层组成;Xt为基于ResNet-18的废旧家电分类网络的输入,定义为该分类网络在第t次分类任务中的特征提取器,θt为的参数矩阵,大小为512×28×28,用分类器作为废旧家电分类网络在第t次分类任务的输出,F为的输入特征,ωt为的参数矩阵,大小为2048×10;3基于对比动态特征增量学习训练废旧家电分类模型基于对比动态特征增量学习训练废旧家电分类模型分为初始训练任务和增量训练任务,训练集Xt分批次输入废旧家电分类网络,经过和完成分类任务,的交叉熵损失函数L为: 其中,为第j类图像的第i张图像数据真实标签,j=1,2,…,10,i=1,2,…,N,N为第j类图像的数目,为图像i属于类别j的预测概率,Y为L的输入图像类别真实标签,P为L的输入图像类别预测概率;①初始训练任务:当t=1时,开始初始训练,训练初始模型参数,第1次分类任务损失函数Lce,1为: 其中,θ1为第1次任务中的参数矩阵,X1为第1次任务的训练图像数据,ω1为的参数矩阵,Y1为X1中图像的类别标签,使用梯度下降算法更新θ1和ω1,更新参数公式为: 其中,为第1次分类任务中,第n+1次迭代训练时的参数矩阵,为n次迭代时的参数矩阵,为第1次分类任务中,第n+1次迭代训练时的参数矩阵,为n次迭代时的参数矩阵,表示偏导;②增量训练任务:当t=2时,训练增量模型参数,在任务t中基于对比动态特征增量学习总损失函数LDP,t公式为:LDP,t=Lce,t+Lμ,t+Lf,t5其中,Lf,t为对比特征损失函数,如公式7,Lμ,t为动态原型分类任务的损失函数,如公式8,第t次分类任务损失函数Lce,t公式为: 其中,Yt为Xt中图像的类别标签,ωt为分类器的参数矩阵,在第t次任务中不同类别对比动态特征损失函数公式Lf,t为: 其中,为任务t中第k类的第q张图像数据,q=1,2,…,Q,Q为第k类图像数目,k=1,2,…,10,j≠k,为任务t中第j类的第i张图像数据,在任务t中应提供第j类的第i张图像动态原型将任务t中的送入分类器得到动态原型分类任务的损失函数表达公式Lμ,t为: 其中,Yt,p为的类标签,由和组成,动态原型公式为: 其中,使用梯度下降算法优化增量废旧家电分类网络中的参数θt和ωt,参数更新公式为: 其中,为第t次分类任务中,第n+1次迭代训练时的参数矩阵,为n次迭代时的参数矩阵,为第t次分类任务中,第n+1次迭代训练时的参数矩阵,θt为n次迭代时的参数矩阵,当第t次增量训练任务模型参数完成更新时,t增加1,开始进行第t+1次增量训练任务,当t=5时,完成模型参数训练;4识别废旧家电类型利用训练完成的基于对比动态特征增量学习废旧家电模型,对测试集中的10类废旧家电图像进行识别,得到废旧家电预测类别标签完成废旧家电识别任务。
全文数据:
权利要求:
百度查询: 北京工业大学 一种基于对比动态特征增量学习的废旧家电识别方法
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。