[发明专利]字符生成模型的训练方法、字符生成方法、装置和设备和介质在审
申请号: | 202111057838.0 | 申请日: | 2021-09-09 |
公开(公告)号: | CN113792526A | 公开(公告)日: | 2021-12-14 |
发明(设计)人: | 唐礼承;刘家铭 | 申请(专利权)人: | 北京百度网讯科技有限公司 |
主分类号: | G06F40/109 | 分类号: | G06F40/109;G06T3/00 |
代理公司: | 北京品源专利代理有限公司 11332 | 代理人: | 高艳红 |
地址: | 100085 北京市*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 字符 生成 模型 训练 方法 装置 设备 介质 | ||
1.一种字符生成模型的训练方法,包括:
将源域样本字和目标域风格字输入至字符生成模型,得到目标域生成字;
将所述目标域生成字和目标域样本字输入至预先训练的字符分类模型,计算所述字符生成模型的特征损失;
根据所述特征损失调整所述字符生成模型的参数。
2.根据权利要求1所述的方法,其中,所述将所述目标域生成字和目标域样本字输入至预先训练的字符分类模型,计算所述字符生成模型的特征损失,包括:
将所述目标域生成字输入到所述字符分类模型,得到所述字符分类模型的至少一个特征层输出的生成特征图;
将所述目标域样本字输入到所述字符分类模型,得到所述字符分类模型的所述至少一个特征层输出的样本特征图;
根据所述至少一个特征层的生成特征图和样本特征图之间的差异,计算所述字符生成模型的特征损失。
3.根据权利要求2所述的方法,其中,所述计算所述字符生成模型的特征损失,包括:
针对所述至少一个特征层中的每一特征层,计算所述特征层的生成特征图和样本特征图之间的像素差异,得到所述特征层的像素损失;
根据所述至少一个特征层的像素损失,计算所述字符生成模型的特征损失。
4.根据权利要求3所述的方法,其中,所述计算所述特征层的生成特征图和样本特征图之间的像素差异,包括:
针对所述特征层的生成特征图中的每个位置的像素点,计算所述像素点的像素值与所述样本特征图中对应位置的像素点的像素值之间的差值的绝对值,得到每个位置的像素点的差异;
根据多个位置的像素点的差异,确定所述特征层的生成特征图和样本特征图之间的像素差异。
5.根据权利要求1所述的方法,还包括:
将所述目标域风格字输入至字符生成模型,得到所述目标域风格字的第一风格特征向量;
将所述目标域生成字输入至所述字符生成模型中,得到所述目标域生成字的第二风格特征向量;
将所述第二风格特征向量和所述第一风格特征向量输入至组件分类模型中,计算组件分类损失;
将所述目标域样本字和所述目标域生成字输入至鉴别模型中,计算字符对抗损失和风格对抗损失;
将所述目标域生成字输入至所述字符分类模型,计算错字损失;
根据所述组件分类损失、所述字符对抗损失、所述风格对抗损失和所述错字损失,调整所述字符生成模型的参数。
6.根据权利要求1至5中任一项所述的方法,其中,所述源域样本字为具有源域字体风格的图像,所述目标域样本字为具有目标域字体风格的图像。
7.一种字符生成方法,包括:
获取源域输入字,和对应的目标域输入字;
将所述源域输入字和所述目标输入字输入到字符生成模型中,得到目标域新字;其中,所述字符生成模型根据如权利要求1至6中任一项所述的字符生成模型的训练方法训练得到。
8.一种字符生成模型的训练装置,包括:
目标域生成字获取模块,用于将源域样本字和目标域风格字输入至字符生成模型,得到目标域生成字;
特征损失计算模块,用于将所述目标域生成字和目标域样本字输入至预先训练的字符分类模型,计算所述字符生成模型的特征损失;
第一损失调整模块,用于根据所述特征损失调整所述字符生成模型的参数。
9.根据权利要求8所述的装置,其中,所述特征损失计算模块,包括:
第一特征图生成单元,用于将所述目标域生成字输入到所述字符分类模型,得到所述字符分类模型的至少一个特征层输出的生成特征图;
第二特征图生成单元,用于将所述目标域样本字输入到所述字符分类模型,得到所述字符分类模型的所述至少一个特征层输出的样本特征图;
特征损失计算单元,用于根据所述至少一个特征层的生成特征图和样本特征图之间的差异,计算所述字符生成模型的特征损失。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京百度网讯科技有限公司,未经北京百度网讯科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202111057838.0/1.html,转载请声明来源钻瓜专利网。