Invention Publication
- Patent Title: 图像生成模型的训练方法、装置及电子设备
-
Application No.: CN202111528315.XApplication Date: 2021-12-14
-
Publication No.: CN116263991APublication Date: 2023-06-16
- Inventor: 贾麒霏 , 闫泽杭 , 张亚森
- Applicant: 北京小米移动软件有限公司 , 北京小米松果电子有限公司
- Applicant Address: 北京市海淀区西二旗中路33号院6号楼8层018号;
- Assignee: 北京小米移动软件有限公司,北京小米松果电子有限公司
- Current Assignee: 北京小米移动软件有限公司,北京小米松果电子有限公司
- Current Assignee Address: 北京市海淀区西二旗中路33号院6号楼8层018号;
- Agency: 北京法胜知识产权代理有限公司
- Agent 戎郑华
- Main IPC: G06V10/44
- IPC: G06V10/44 ; G06V10/24 ; G06V10/774 ; G06V10/74 ; G06V10/82 ; G06N3/0464

Abstract:
本公开提供一种图像生成模型的训练方法、装置及电子设备,其中,方法包括:通过获取原始图像,以及对应的残缺图像,将残缺图像输入图像生成模型,以对残缺图像中残缺的局部区域进行修正,得到修正图像。在对原始图像进行特征提取以得到第一特征矩阵,以及对修正图像进行特征提取以得到第二特征矩阵之后,对第一特征矩阵进行特征分解以得到第一分解特征信息,以及对第二特征矩阵进行特征分解以得到第二分解特征信息。进而根据第一分解特征信息和第二分解特征信息之间的相似度,对图像生成模型进行训练,得到训练完成的图像生成模型。由此,通过基于分解特征信息对图像生成模型进行训练,促使图像生成模型关注到不同分解特征,使图像整体保持协调。
Information query