- 专利标题: 图像生成模型的训练方法及风格化图像生成方法
-
申请号: CN202310443477.6申请日: 2023-04-24
-
公开(公告)号: CN116152901B公开(公告)日: 2023-08-01
- 发明人: 张顺四 , 黄深能 , 张强
- 申请人: 广州趣丸网络科技有限公司
- 申请人地址: 广东省广州市天河区平云路163号之二1601室
- 专利权人: 广州趣丸网络科技有限公司
- 当前专利权人: 广州趣丸网络科技有限公司
- 当前专利权人地址: 广东省广州市天河区平云路163号之二1601室
- 代理机构: 北京集佳知识产权代理有限公司
- 代理商 任文生
- 主分类号: G06V40/16
- IPC分类号: G06V40/16 ; G06V10/74 ; G06V10/774 ; G06V10/82 ; G06N3/08
摘要:
本申请提供了一种图像生成模型的训练方法及风格化图像生成方法,该训练方法采用对比学习的方式训练图像生成模型。在每轮次训练过程中,计算机设备可根据图像生成模型输出的样本风格化图像、对比学习模型输出的各张正例风格化图像和各张负例风格化图像,计算当前训练轮次对应的训练损失值,并据此训练图像生成模型的参数权重。并且,在每轮次训练过程中,计算机设备可根据样本风格化图像、各张正例风格化图像、各张负例风格化图像、图像生成模型在当前训练轮次对应的调整前后的参数权重,动态自适应地更新对比学习网络的参数权重,从而可根据样本学习的难易程度自动调整模型的更新速度,进而能有效加速图像生成模型的收敛,缩减训练耗时。
公开/授权文献
- CN116152901A 图像生成模型的训练方法及风格化图像生成方法 公开/授权日:2023-05-23