一种面向多人姿态估计应用的遮挡数据集构造方法

    公开(公告)号:CN118823507A

    公开(公告)日:2024-10-22

    申请号:CN202410827342.4

    申请日:2024-06-25

    申请人: 郑州大学

    摘要: 本发明公开了一种面向多人姿态估计应用的遮挡数据集构造方法,包括以下步骤:S1,从非遮挡数据集中筛选单人图像,使用人像抠图算法对筛选后的单人图像进行抠图并根据人体框标注信息进行剪裁,同时更新关键点位置信息,剪裁后的单人图像及其更新后的标注数据构成单人数据集;S2,从非遮挡数据集中选取一副图像作为待遮挡图像,然后从构建的单人数据集中随机抽取一幅单人图像缩放后对待遮挡图像中的人体进行定位遮挡,同时在非遮挡数据集的标注文件中添加单人图像的人体标注信息,生成遮挡图像及其对应的标注文件;S3,通过对非遮挡数据集中的所有图像进行遮挡操作,构造带标注信息的遮挡数据集。本发明通过人像抠图和定位遮挡的方法构造遮挡数据集,能够减少人工采集图像成本以及减小对遮挡数据集的标注难度,同时能够提高模型在遮挡场景下人体姿态估计性能。