公开(公告)号：CN119228937B

公开(公告)日：2025-04-01

申请号：CN202411730361.1

申请日：2024-11-28

Applicant: 杭州海康威视数字技术股份有限公司

Inventor： 郑海涛 ,  彭怀亮

IPC: G06T11/00 ,  G06T5/60 ,  G06T9/00 ,  G06N3/0455 ,  G06N3/084

Abstract: 本申请实施例提供了一种图像生成模型的训练方法及图像生成方法，涉及图像处理技术领域。该训练方法包括：将第一样本图像输入图像生成模型，使图像生成模型对第一样本图像进行编码，基于得到的图像特征向量，利用干净图像先验码本和彩虹纹图像先验码本分别预测干净图像编码和彩虹纹图像编码，对干净图像编码和彩虹纹图像编码对比所得到的背景层编码和彩虹纹层编码进行编码融合，对得到的目标彩虹纹图像编码进行解码，得到彩虹纹图像；基于第一样本图像和彩虹纹图像，计算第一损失值，基于第一损失值，调整图像生成模型的模型参数。以上可见，应用本申请实施例提供的方案，可以训练得到的图像生成模型，以生成具有较高保真度的彩虹纹车窗图像。

公开(公告)号：CN119228937A

公开(公告)日：2024-12-31

申请号：CN202411730361.1

申请日：2024-11-28

Applicant: 杭州海康威视数字技术股份有限公司

Inventor： 郑海涛 ,  彭怀亮

IPC: G06T11/00 ,  G06T5/60 ,  G06T9/00 ,  G06N3/0455 ,  G06N3/084

Abstract: 本申请实施例提供了一种图像生成模型的训练方法及图像生成方法，涉及图像处理技术领域。该训练方法包括：将第一样本图像输入图像生成模型，使图像生成模型对第一样本图像进行编码，基于得到的图像特征向量，利用干净图像先验码本和彩虹纹图像先验码本分别预测干净图像编码和彩虹纹图像编码，对干净图像编码和彩虹纹图像编码对比所得到的背景层编码和彩虹纹层编码进行编码融合，对得到的目标彩虹纹图像编码进行解码，得到彩虹纹图像；基于第一样本图像和彩虹纹图像，计算第一损失值，基于第一损失值，调整图像生成模型的模型参数。以上可见，应用本申请实施例提供的方案，可以训练得到的图像生成模型，以生成具有较高保真度的彩虹纹车窗图像。