利索能及
我要发布
收藏
专利号: 2020113965200
申请人: 北京百度网讯科技有限公司
专利类型:发明专利
专利状态:授权未缴费
更新日期:2024-07-30
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种图像生成模型训练方法,所述方法包括:

获取预先建立的生成式对抗网络,其中,所述生成式对抗网络包括:生成器和判别器,所述生成器由编码器和解码器构成,所述编码器采用卷积神经网络从输入的图像中提取特征;所述解码器利用反卷积层完成从特征向量中还原出低级特征的工作,得到生成图像;所述解码器包括多个串联连接的生成模块,所述多个生成模块用于对输入的图像进行不同倍率的分辨率提升处理,以得到多个不同尺度图像;所述判别器包括多个不同尺度的判别模块,各个判别模块分别独立;

执行以下训练步骤:从预置的样本集中选取样本,每个样本包括第一域图像和多个不同尺度的第二域图像;将该样本的第一域图像输入所述生成式对抗网络,得到各个生成模块输出的不同尺度的第二域的伪图像;针对各个伪图像,将该伪图像以及该伪图像对应尺度的第二域图像一起输入对应尺度的判别模块,计算所述生成式对抗网络的损失值;响应于确定所述生成式对抗网络满足训练完成条件,则将所述生成式对抗网络作为图像生成模型。

2.根据权利要求1所述的方法,其中,各个生成模块的输出图像的尺度是输入图像的2倍。

3.根据权利要求1所述的方法,其中,所述生成式对抗网络还包括:多个不同尺度的上采样模块;

各个上采样模块与其对应尺度的生成模块并联连接,且与各个上采样模块并联的生成模块不包括所述多个生成模块中的首部和尾部的生成模块。

4.根据权利要求1所述的方法,其中,所述计算所述生成式对抗网络的损失值包括:计算各个判别模块的损失值;

对所有判别模块的损失值求均值,得到平均损失值,将所述平均损失值作为所述生成式对抗网络的损失值。

5.根据权利要求1所述的方法,其中,所述计算所述生成式对抗网络的损失值包括:计算各个判别模块的损失值;

获取各个判别模块的尺度权重值;

将各个判别模块的损失值与其对应的尺度权重值相乘,得到各个判别模块的权重损失值;

对所有判别模块的权重损失值求均值,得到所述生成式对抗网络的损失值。

6.一种图像生成方法,所述方法包括:

获取待处理图像;

将所述待处理图像输入采用如权利要求1‑5中任一项所述的方法生成的图像生成模型中,输出生成后的图像。

7.一种图像生成模型训练装置,所述装置包括:

网络建立单元,被配置成获取预先建立的生成式对抗网络,其中,所述生成式对抗网络包括:生成器和判别器,所述生成器由编码器和解码器构成,所述编码器采用卷积神经网络从输入的图像中提取特征;所述解码器利用反卷积层完成从特征向量中还原出低级特征的工作,得到生成图像;所述解码器包括多个串联连接的生成模块,所述多个生成模块用于对输入的图像进行不同倍率的分辨率提升处理,以得到多个不同尺度图像;所述判别器包括多个不同尺度的判别模块,各个判别模块分别独立;

样本选取单元,被配置成从预置的样本集中选取样本,每个样本包括第一域图像和多个不同尺度的第二域图像;

样本输入单元,被配置成将该样本的第一域图像输入所述生成式对抗网络,得到各个生成模块输出的不同尺度的第二域的伪图像;

图像输入单元,被配置成针对各个伪图像,将该伪图像以及该伪图像对应尺度的第二域图像一起输入对应尺度的判别模块,计算所述生成式对抗网络的损失值;

模型输出单元,被配置成响应于确定所述生成式对抗网络满足训练完成条件,则将所述生成式对抗网络作为图像生成模型。

8.根据权利要求7所述的装置,其中,各个生成模块的输出图像的尺度是输入图像的2倍。

9.根据权利要求7所述的装置,其中,所述生成式对抗网络还包括:多个不同尺度的上采样模块;

各个上采样模块与其对应尺度的生成模块并联连接,且与各个上采样模块并联的生成模块不包括所述多个生成模块中的首部和尾部的生成模块。

10.根据权利要求7所述的装置,其中,所述图像输入单元包括:计算模块,被配置成计算各个判别模块的损失值;

均值模块,被配置成对所有判别模块的损失值求均值,得到平均损失值,将所述平均损失值作为所述生成式对抗网络的损失值。

11.根据权利要求7所述的装置,其中,所述图像输入单元包括:计损模块,被配置成计算各个判别模块的损失值;

获取模块,被配置成获取各个判别模块的尺度权重值;

相乘模块,被配置成将各个判别模块的损失值与其对应的尺度权重值相乘,得到各个判别模块的权重损失值;

得到模块,被配置成对所有判别模块的权重损失值求均值,得到所述生成式对抗网络的损失值。

12.一种图像生成装置,所述装置包括:

图像获取单元,被配置成获取待处理图像;

图像输出单元,被配置成将所述待处理图像输入采用如权利要求1‑5中任一项所述的方法生成的图像生成模型中,输出生成后的图像。

13.一种电子设备,其特征在于,包括:

至少一个处理器;以及

与所述至少一个处理器通信连接的存储器;其中,

所述存储器存储有可被所述至少一个处理器执行的指令,所述指令被所述至少一个处理器执行,以使所述至少一个处理器能够执行权利要求1‑6中任一项所述的方法。

14.一种存储有计算机指令的非瞬时计算机可读存储介质,其特征在于,所述计算机指令用于使所述计算机执行权利要求1‑6中任一项所述的方法。