[发明专利]一种基于深度学习的人头部图像提取方法和装置

摘要

本发明公开了一种基于深度学习的人头部图像提取方法和装置。所述方法包括将包含人像的待提取图片输入到经过预先训练的YOLO神经网络进行处理和输出头部区域图像，将头部区域图像输入到经过训练的HED神经网络进行处理和输出头部边缘轮廓图像，将头部区域图像作为内容图像、头部边缘轮廓图像作为风格图像输入到艺术风格迁移神经网络进行处理，从而输出合成图像等步骤；所述装置包括存储器和处理器。本发明能够从头部区域图像中提取出头部边缘轮廓图像，并将头部边缘轮廓图像的风格迁移到头部区域图像中，实现头部边缘轮廓、五官和头发等细节的融合，使得合成图像具有丰富的面部特征和清晰的边缘轮廓。本发明应用于图像识别处理技术领域。

主权项

1.一种基于深度学习的人头部图像提取方法，其特征在于，包括以下步骤：S1.将包含人像的待提取图片输入到经过预先训练的YOLO神经网络进行处理，从而输出头部区域图像，所述头部区域图像为待提取图片的一部分且包含所述人像的头像；S2.将头部区域图像输入到经过训练的HED神经网络进行处理，从而输出头部边缘轮廓图像；S3.将头部区域图像作为内容图像、头部边缘轮廓图像作为风格图像输入到艺术风格迁移神经网络进行处理，从而输出合成图像，所述艺术风格迁移神经网络用于将头部边缘轮廓图像的风格迁移到头部区域图像。