上下文增强的图像生成方法、模型训练方法及系统

• 国民经济行业：信息传输、软件和信息技术服务业／互联网和相关服务／互联网信息服务
• 学科类别：电子、通信与自动控制技术／信息处理技术／图象处理
• 权利人：上海交通大学

知识产权简介

本发明提供一种上下文增强的图像生成模型训练方法及图像生成方法、系统，包括：获取训练图像数据集；利用预设的加噪机制对训练图像进行多步加噪，每一步所加的高斯噪声的方差取决于当前步数并逐渐增大，直至将未加噪的训练图像转换为标准高斯噪声，得到各步的加噪图像；利用随机掩码机制生成掩码，并根据掩码对加噪图像进行掩码区域的丢弃和对未加噪图像进行未掩码区域的丢弃；采用未丢弃的加噪图像和未加噪图像作为输入，对上下文增强的图像生成模型进行训练，预测每一步的加噪图像，去除噪声逐步恢复未加噪图像，用于图像生成。本发明能有效提升图像生成方法的上下文理解能力，提高图像生成质量，实现高分辨率多样性的图像生成。