公开(公告)号：CN117372506A

公开(公告)日：2024-01-09

申请号：CN202311314357.2

申请日：2023-10-11

Applicant: 南京邮电大学

Inventor： 叶宁 ,  张强 ,  肖甫 ,  郭政鑫 ,  桂林卿 ,  徐康 ,  王甦 ,  汪莹

IPC: G06T7/70 ,  G06N3/0475 ,  G06N3/094 ,  G06N3/0495 ,  G06N3/082 ,  G06N3/0464 ,  G06N3/045

Abstract: 本发明涉及一种基于知识蒸馏和生成对抗网络的面向边缘设备的姿态估计方法，属于图像识别技术领域；主要包括：将处理后的训练集输入到Hourglass网络中训练教师模型和学生模型；基于生成对抗网络的知识蒸馏方法，使用原始教师模型的输出结果作为真值获得对抗损失，同时结合原始学生模型自身的损失对原始学生模型进行再训练，获得性能更加优异的学生模型；进行新学生模型部署；与传统的知识蒸馏方法相比，本发明能够更加有效地将教师模型的知识传递给学生模型，使学生模型能够更加有效的获取教师模型的知识，解决当前姿态估计方案难以部署到边缘设备上的问题。