一种基于仿射变换的自监督人体关键点检测方法

    公开(公告)号:CN117636389A

    公开(公告)日:2024-03-01

    申请号:CN202311545798.3

    申请日:2023-11-20

    Abstract: 本发明公开了一种基于仿射变换的自监督人体关键点检测方法,包括如下步骤:S1.采集得到不同人物、不同姿态的行人图像及视频数据集,并对数据集进行预处理操作;S2.预定义人体关键点模板;S3.构建卷积神经网络;S4.通过编码器网络对行人图像样本进行姿态特征提取,利用姿态特征生成仿射矩阵;S5.使用仿射矩阵对关键点模板进行仿射变换,得到与行人图像样本中的人体姿态相对应的关键点,并设置边沿损失对仿射矩阵的生成进行约束;S6.将检测到的人体关键点按预定义的语义信息进行一一连接,生成人体骨架热图;S7.将随机掩膜后的行人图像样本与其对应的人体骨架热图在通道维度上进行拼接后馈送至解码器,利用解码器修复掩膜后的图像样本,得到修复的图像。

Patent Agency Ranking