[发明专利]用于生成眼底图像生成模型的方法和装置

摘要

本公开的实施例公开了用于生成眼底图像生成模型的方法和装置。该方法的一具体实施方式包括：获取训练样本集合；获取初始生成对抗网络；从训练样本集合中选取训练样本，以及执行以下训练步骤：将选取的训练样本的第一类样本眼底图像输入初始生成对抗网络，得到第一输出眼底图像和第二输出眼底图像；分析确定第一判别损失值、第二判别损失值和图像差异；分析确定训练样本的总损失值，以及将训练样本的总损失值与目标值进行比较；根据比较结果确定第一生成网络是否训练完成；响应于确定第一生成网络训练完成，将第一生成网络确定为眼底图像生成模型。该实施方式有助于提高利用眼底图像生成模型生成眼底图像的真实度。

主权项

1.一种用于生成眼底图像生成模型的方法，包括：获取训练样本集合，其中，训练样本包括含有预定类型图像区域的第一类样本眼底图像和不含有预定类型图像区域的第二类样本眼底图像；获取初始生成对抗网络，其中，所述初始生成对抗网络包括第一生成网络、第二生成网络、第一判别网络和第二判别网络；从所述训练样本集合中选取训练样本，以及执行以下训练步骤：将选取的训练样本的第一类样本眼底图像输入第一生成网络，得到不含有预定类型图像区域的第一输出眼底图像；将第一输出眼底图像输入第二生成网络，得到含有预定类型图像区域的第二输出眼底图像；将第一输出眼底图像和选取的训练样本的第二类样本眼底图像输入第一判别网络，得到第一判别损失值；将第二输出眼底图像和选取的训练样本的第一类样本眼底图像输入第二判别网络，得到第二判别损失值；对第二输出眼底图像与选取的训练样本的第一类样本眼底图像进行图像差异分析，确定图像差异；根据预设第一判别权重、预设第二判别权重和预设图像重构权重，将第一判别损失值、第二判别损失值和图像差异的加权结果作为训练样本的总损失值，以及将训练样本的总损失值与目标值进行比较；根据比较结果确定第一生成网络是否训练完成；响应于确定第一生成网络训练完成，将第一生成网络确定为眼底图像生成模型。