[发明专利]一种动漫头像生成方法和系统

说明书

本发明提供了一种动漫头像生成方法和系统，包括：对原始图像进行处理，得到预处理图像；所述原始图像为用户上传的真实图像；将所述预处理图像输入图像识别模型，模型输出第一标签；将所述预处理图像和所述第一标签输入标签纠正模型，模型输出第二标签；所述第二标签通过纠正第一标签得到；将所述第二标签和所述预处理图像输入动漫头像生成模型，模型输出动漫头像；其中，模型输出动漫头像为动漫头像生成模型多次将噪声引入所述预处理图像并采样，得到采样图像，将预设次数的所述采样图像作为所述动漫头像输出；以使得处理得到的动漫头像达到与输入图像相似的同时又保证生成的图像是动漫风格。

技术领域

本发明涉及图像处理技术领域，具体而言，涉及一种动漫头像生成方法和系统。

背景技术

随着互联网的发展，人们将日常生活的点点滴滴记录在互联网上进行分享，通过在网络上展示图片可以更好的记录当下的生活，对图片进行处理就应运而生，为了更好的为用户提供个性化展示，图片处理包括多种风格，动漫风格处理属于其中一种。传统的真实图像转动漫的处理需要使用虚拟摄像机进行拍摄，并对拍摄数据进行图像渲染处理，以得到动漫头像。因此，现有图像动漫处理的方式存在与输入图像相似度过高，图像转换效果不明显的问题，而且需要的转换设备较多，例如，需要特殊摄像头等专业设备；且难以做到与输入图像相似的同时又保证生成的图像是动漫风格。

有鉴于此，本发明提出了一种动漫头像生成方法和系统，以使得处理得到的动漫头像达到与输入图像相似的同时又保证生成的图像是动漫风格。

发明内容

本发明的目的在于提供一种动漫头像生成方法，包括：对原始图像进行处理，得到预处理图像；所述原始图像为用户上传的真实图像；将所述预处理图像输入图像识别模型，模型输出第一标签；将所述预处理图像和所述第一标签输入标签纠正模型，模型输出第二标签；所述第二标签通过纠正第一标签得到；将所述第二标签和所述预处理图像输入动漫头像生成模型，模型输出动漫头像；其中，模型输出动漫头像为动漫头像生成模型多次将噪声引入所述预处理图像并采样，得到采样图像，将预设次数的所述采样图像作为所述动漫头像输出。

进一步的，所述图像识别模型通过模型训练得到，包括：将第一训练样本输入初始图像识别模型；所述第一训练样本包括动漫人脸图像和对应的第一描述文件；所述初始图像识别模型为deepdanbooru模型；基于所述初始图像识别模型的输出和所述第一描述文件构建第一损失函数；基于所述第一损失函数迭代求解所述初始图像识别模型的参数，将训练完成的初始图像识别模型作为所述图像识别模型。

进一步的，所述将所述预处理图像和所述第一标签输入标签纠正模型，模型输出第二标签，包括：将所述预处理图像输入描述模型，模型输出对所述预处理图像的自然语言描述；将所述自然语言描述与所述第一标签做比对，纠正部分标签结果。

进一步的，所述描述模型通过模型训练得到，包括：将第二样本输入初始描述模型；所述第二样本包括真实人脸图像和对应的第二描述文件；所述初始描述模型为clip模型；基于所述初始描述模型的输出和所述第二描述文件构建第二损失函数；基于所述第二损失函数迭代求解所述初始描述模型的参数，将训练完成的初始描述模型作为所述描述模型。

进一步的，所述动漫头像生成模型通过模型训练得到，包括：将第一训练样本输入初始动漫头像生成模型；所述第一训练样本包括动漫人脸图像和对应的第一描述文件；所述初始动漫头像生成模型为Anything模型；基于所述初始动漫头像生成模型的输出和所述动漫人脸图像构建第三损失函数；基于所述第三损失函数迭代求解所述初始动漫头像生成模型的参数，将训练完成的初始动漫头像生成模型作为所述动漫头像生成模型。