基于知识蒸馏的轻量级人体姿态估计方法及装置

本发明公开了基于知识蒸馏的轻量级人体姿态估计方法及装置，涉及人体姿态估计领域，方法包括：S1，构建初始数据集，对数据集进行标记；S2，构建用于人体姿态估计的教师模型和学生模型；S3，使用标记的数据集对教师模型进行训练，利用训练好的教师模型对学生模型进行指导训练，得到训练好的学生模型；指导训练包括特征蒸馏阶段和逻辑蒸馏阶段；所述特征蒸馏基于学生模型的中间特征图和剪枝后的教师模型中间特征图；逻辑蒸馏基于教师模型和学生模型的逻辑输出；S4，获取视频数据，将视频数据输入训练好的学生模型进行人体姿态估计。本发明通过引入秩引导的滤波器剪枝与特征关联性重构，降低人体姿态估计模型参数的同时，确保了模型的高效和精确。