首页 专利交易 科技果 科技人才 科技服务 国际服务 商标交易 会员权益 IP管家助手 需求市场 关于龙图腾
 /  免费注册
到顶部 到底部
清空 搜索

一种基于知识蒸馏的疼痛强度估计方法 

买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!

申请/专利权人:齐鲁工业大学(山东省科学院);山东省人工智能研究院

摘要:本发明涉及一种基于知识蒸馏的疼痛强度估计方法,属于表情识别技术领域。其包括以下步骤:获取脸部图像序列;构建基于知识蒸馏的疼痛强度估计模型,包括教师网络和学生网络,分别为集成注意力模块的ResNet50和集成注意力模块的ResNet18,将图像序列分别输入教师网络和学生网络进行图像分析;教师网络和学生网络的各层之间进行知识蒸馏;教师网络和学生网络最终的输出结果进行知识蒸馏;根据模型总损失训练得到最终基于知识蒸馏的疼痛强度估计模型。本发明通过嵌入注意力模块能够减少无关噪音的影响,通过层知识蒸馏进而提升学生网络特征提取能力,从而有效提高模型对于疼痛强度识别的准确性。

主权项:1.一种基于知识蒸馏的疼痛强度估计方法,其特征在于,包括以下步骤:S1.获取原始图像序列,并对原始图像序列进行预处理,得到脸部图像序列,并将所述预处理后的脸部图像序列划分为训练集和测试集;S2.构建基于知识蒸馏的疼痛强度估计模型,所述模型包括教师网络和学生网络,所述教师网络为集成注意力模块的ResNet50网络架构,所述学生网络为集成注意力模块的ResNet18网络架构,所述教师网络和学生网络各有五个网络层,所述教师网络和学生网络的前三个网络层后面各有一个注意力模块att1,所述教师网络和学生网络的后两个网络层的后面各有一个注意力模块att2,将所述训练集中的脸部图像序列分别输入所述教师网络和学生网络进行图像分析;S3.对所述教师网络和学生网络的各层之间进行知识蒸馏,得到各网络层的层蒸馏损失;S4.所述教师网络和学生网络最终的输出结果进行知识蒸馏,得到输出结果蒸馏总损失;S5.将所述层蒸馏损失和输出结果蒸馏总损失相加得到模型总损失,根据模型总损失训练得到最终基于知识蒸馏的疼痛强度估计模型,测试集中的待检测图像经过最终基于知识蒸馏的疼痛强度估计模型进行预测,得到疼痛强度估计结果。

全文数据:

权利要求:

百度查询: 齐鲁工业大学(山东省科学院) 山东省人工智能研究院 一种基于知识蒸馏的疼痛强度估计方法

免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。