本发明属于图像处理领域,具体涉及一种基于目标检测技术的人体边缘提取方法。
背景技术:
1、人体边缘提取即将图像中所有人体的边缘提取出来,显示为人体的轮廓线条,可应用于作战增强现实眼镜设备、无人机侦查设备中,对当前场景中人体进行发现,同时对人体的轮廓进行描绘和提示,能够帮助作战人员发现人体目标,同时对作战或指挥人员产生较少的视觉干扰。目前,目标检测技术能够将场景中的人体通过方框的形式框出,但方形框标注对人的视觉影响较大,视觉效果较差。因此,需要对人体边缘进行进一步提取,人体轮廓边缘和图像中的人体结合显示,突出人体边缘,视觉效果好。目前,图像抠图算法能够实现对场景中显著的人体进行抠图,但是无法检测到在图像场景中后方、不显著的人体,全局人体检测能力较差;目标检测算法能够对场景中的所有人体进行检测,并以方框的形式进行标注,但无法获取人体的边缘信息。
技术实现思路
1、(一)要解决的技术问题
2、本发明要解决的技术问题是如何提供一种基于目标检测技术的人体边缘提取方法,以解决图像抠图算法能够实现对场景中显著的人体进行抠图,但是无法检测到在图像场景中后方、不显著的人体,全局人体检测能力较差;目标检测算法能够对场景中的所有人体进行检测,并以方框的形式进行标注,但无法获取人体的边缘信息的问题。
3、(二)技术方案
4、为了解决上述技术问题,本发明提出一种基于目标检测技术的人体边缘提取方法,该方法包括如下步骤:
5、s1、人体目标检测
6、输入多人人体图像或视频帧,使用人体检测模型对图像中的人体进行检测,得到图像中的人体框位置坐标,形成人体序号和位置坐标的五元组,表示为(pnumber,x1,y1,x2,y2),其中pnumber为图像中检测到人体的序号,x1,y1分别为对应人体框左上角位置的横坐标和纵坐标,x2,y2分别为对应人体框右下角位置的横坐标和纵坐标;
7、s2、多进程人体抠图处理
8、根据检测到的人体框坐标对图像进行裁剪,得到图像中全部人体的n张裁剪图像,确保裁剪后的人体图像中的主要目标为人体,送入人体图像抠图算法进行抠图处理,人体图像抠图算法获得图像中人体部分的遮罩图α;
9、s3、遮罩边缘提取
10、使用canny边缘提取算法对人体遮罩图α进行边缘提取,得到人体部分的边缘图结果ioutput;
11、s4、合并边缘提取结果
12、创建一张和原始多人体输入图像相同大小的黑色图像,将步骤s3中得到边缘图结果ioutput根据对应的左上角和右下角的坐标,替换为黑色图像中对应区域像素,得到处理后的多人体边缘结果图。
13、(三)有益效果
14、本发明提出一种基于目标检测技术的人体边缘提取方法,本发明提出的技术方案中将目标检测算法和图像抠图算法结合用来对图像中的人体边缘进行提取,提升了人体边缘提取算法的完备性和准确性。
1.一种基于目标检测技术的人体边缘提取方法,其特征在于,该方法包括如下步骤:
2.如权利要求1所述的基于目标检测技术的人体边缘提取方法,其特征在于,所述步骤s1中,人体检测模型为yolov7预训练模型。
3.如权利要求1所述的基于目标检测技术的人体边缘提取方法,其特征在于,所述s2中,人体图像抠图算法为matting模型。
4.如权利要求3所述的基于目标检测技术的人体边缘提取方法,其特征在于,所述s2中,裁剪后的图片作为多进程人体抠图过程的输入,根据检测人体数量创建进程,进行人体图像抠图算法模型推理。
5.如权利要求4所述的基于目标检测技术的人体边缘提取方法,其特征在于,所述s2具体处理过程为:
6.如权利要求1-5任一项所述的基于目标检测技术的人体边缘提取方法,其特征在于,所述步骤s3中,使用边缘差分sobel算子计算差分的水平方向gx和垂直方向的gy,
7.如权利要求6所述的基于目标检测技术的人体边缘提取方法,其特征在于,所述步骤s3中,遮罩图α仅有黑白两种像素,不进行图像去噪,直接计算遮罩图α的图像梯度进行边缘提取。
8.如权利要求1所述的基于目标检测技术的人体边缘提取方法,其特征在于,将人体的边缘图结果和图像中的人体进行组合显示。
9.如权利要求8所述的基于目标检测技术的人体边缘提取方法,其特征在于,该方法应用到ar眼镜中的图像分析显示模块。
10.如权利要求8所述的基于目标检测技术的人体边缘提取方法,其特征在于,该方法应用到人体目标检测设备中的图像分析显示模块。