← 返回列表
生成对抗网络的训练方法、双向图像风格转换方法和装置
申请人信息
- 申请人:南京信息工程大学
- 申请人地址:210044 江苏省南京市江北新区宁六路219号
- 发明人: 南京信息工程大学
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 生成对抗网络的训练方法、双向图像风格转换方法和装置 |
| 专利类型 | 发明申请 |
| 申请号 | CN202410101475.3 |
| 申请日 | 2024/1/25 |
| 公告号 | CN117635418A |
| 公开日 | 2024/3/1 |
| IPC主分类号 | G06T3/04 |
| 权利人 | 南京信息工程大学 |
| 发明人 | 朱高峰; 瞿治国; 孙乐 |
| 地址 | 江苏省南京市宁六路219号 |
摘要文本
本发明公开了一种生成对抗网络的训练方法、双向图像风格转换方法和装置，所述生成对抗网络的训练方法包括：获取用于训练的图像数据集，其中所述图像数据集包括多个真实第一风格图像以及对应的真实第二风格图像；利用所述图像数据集对生成对抗网络进行训练，获得训练好的生成对抗网络。该生成对抗网络的训练方法、双向图像风格转换方法和装置，通过在生成网络中加入特征二维注意力模块，并与卷积层、转置卷积层和反卷积层相配合，根据目标域的特征对图像进行自适应的调整，以在训练时采用虚拟批正则化和的手段提升训练速度和收敛速度，从而根据训练后的生成对抗网络模拟现有的风格创作出一幅全新的相似该风格的图像。
专利主权项内容
1.一种生成对抗网络的训练方法，其特征是，包括：获取用于训练的图像数据集，其中所述图像数据集包括多个真实第一风格图像以及对应的真实第二风格图像；利用所述图像数据集对生成对抗网络进行训练，获得训练好的生成对抗网络；其中，所述生成对抗网络包括生成网络和判别网络，训练过程具体包括：将真实第一风格图像输入生成网络，得到生成第二风格图像；将生成第二风格图像输入判别网络，得到生成第二风格图像的判别结果；基于所述生成第二风格图像与真实第二风格图像以及生成第二风格图像的判别结果，利用预构建的生成对抗网络的损失函数，进行迭代训练，直至达到预设条件，得到训练好的生成对抗网络。