面向仰卧起坐的多头注意力姿态估计方法及检测系统

    公开(公告)号:CN115953834A

    公开(公告)日:2023-04-11

    申请号:CN202211622338.1

    申请日:2022-12-16

    Inventor: 许国良 汪子璇

    Abstract: 本发明涉及计算机视觉领域,公开了一种面向仰卧起坐的多头注意力姿态估计方法及检测系统,包括:S1、获取若干关于仰卧起坐的原始图像构建自建数据集并进行标注,然后使用合成数据集将若干所述原始图像进行扩充得到扩充数据集;S2、构建多头自注意力模块Mx和ResNet50神经网络,并将所述多头自注意力模块Mx添加至ResNet50神经网络中,得到用于提取图像特征的特征提取网络;S3、构建基于人体坐标回归的人体姿态估计器Ex和损失函数Lloss,将所述特征提取网络和人体姿态估计器Ex通过平均池化层连接回归得到人体姿态估计网络;S4、使用训练完后的所述面向仰卧起坐的多头注意力姿态估计网络对所述测试集中的图像进行识别。本发明可以避免发生过拟合现象,提高图像全局信息的利用,减少推理内存。

Patent Agency Ranking