一种高效的人体姿态估计方法及其模型搭建方法

专利主权项内容

1.一种高效的人体姿态估计模型的搭建方法，其特征在于，包括以下步骤：步骤1，获取人体姿态估计的原始数据；所述原始数据通过公用数据集网站下载或者自拍获取；步骤2，对步骤1中的原始数据进行预处理，并划分训练集和测试集；步骤3，基于HRNet网络结构，搭建一种轻量化的人体姿态估计模型GSEHRNet网络模型；基于高分辨率的网络，通过并行连接的方式，不断融合不同分辨率分支的特征；首先通过卷积层下采样，然后通过第一层Layer1模块，然后通过三个过渡层Transition结构以及三个特征提取阶段Stage，每通过一个过渡层Transition结构就会新增一个尺度分支，以此来构建多分支的网络结构；后面每个特征提取阶段Stage通过这种多分支的并行结构不断的提取特征信息，并对不同分支的不同分辨率的特征信息进行融合，以提取更多的特征信息；最后再将上述提取后的信息的最高分辨率分支的特征信息作为输出；步骤4，使用步骤2中所述的训练集对步骤3中所搭建的GSEHRNet网络模型进行训练；步骤5，使用步骤2中所述的测试集对训练后的GSEHRNet网络模型进行测试，从而获取最终模型。