[发明专利]一种用于将视网膜眼底图像在不同图像域中迁移的方法、电子设备及计算机可读存储介质

权利要求书

1.一种用于将视网膜眼底图像在不同图像域中迁移的方法，其特征在于，包括：

获取被表示在第一图像域中、待被迁移到第二图像域的眼底图像，所述第一图像域与所述第二图像域具有不同的成像参数；以及

将所述眼底图像输入循环生成对抗网络的一生成器，以生成被表示在所述第二图像域中的目标图像，所述循环生成对抗网络由镜像对称的两个生成对抗网络组合形成并且所述生成器为所述两个生成对抗网络经对抗训练而得到的最优生成器，

其中，所述循环生成对抗网络包括第一生成器和第二生成器以及第一鉴别器和第二鉴别器，构建所述循环生成对抗网络包括：

接收训练图像集，所述训练图像集包括被表示在所述第一图像域中的第一眼底图像集和被表示在所述第二图像域中的第二眼底图像集；以及

利用所述两个生成对抗网络和用于对眼底图像分割视杯视盘的目标检测器，对所述第一眼底图像集和所述第二眼底图像集进行处理，以优化所述第一生成器、所述第二生成器、所述第一鉴别器和所述第二鉴别器。

2.根据权利要求1所述的方法，其特征在于，其中对所述第一眼底图像集和所述第二眼底图像集进行处理包括：

将所述第一眼底图像集中的第一眼底图像输入由所述第一生成器、所述第二生成器和所述第一鉴别器构成的第一生成对抗网络以对所述第一眼底图像进行处理，并且将所述第二眼底图像集中的第二眼底图像输入由所述第二生成器、所述第一生成器和所述第二鉴别器构成的第二生成对抗网络以对所述第二眼底图像进行处理；

计算所述第一生成对抗网络的第一鉴别器损失函数和第一生成器损失函数以及所述第二生成对抗网络的第二鉴别器损失函数和第二生成器损失函数；以及

基于所述第一鉴别器损失函数和第一生成器损失函数以及所述第二鉴别器损失函数和第二生成器损失函数，调整所述第一生成器、所述第二生成器、所述第一鉴别器和所述第二鉴别器的参数。

3.根据权利要求2所述的方法，其特征在于，其中将所述第一眼底图像输入所述第一生成对抗网络进行处理包括：

通过所述第一生成器将所述第一眼底图像转换为第一目标图像，所述第一目标图像为所述第一眼底图像在所述第二图像域中的表示；

通过所述第二生成器将所述第一目标图像转换为第一重构图像，所述第一重构图像为所述第一目标图像在所述第一图像域中的表示；以及

通过所述第一鉴别器对所述第一目标图像和所述第二眼底图像进行鉴别，以判断所述第一目标图像是否被真实表示在所述第二图像域中。

4.根据权利要求3所述的方法，其特征在于，其中将所述第二眼底图像输入所述第二生成对抗网络进行处理包括：

通过所述第二生成器将所述第二眼底图像转换为第二目标图像，所述第二目标图像为所述第二眼底图像在所述第一图像域中的表示；

通过所述第一生成器将所述第二目标图像转换为第二重构图像，所述第二重构图像为所述第二目标图像在所述第二图像域中的表示；以及

通过所述第二鉴别器对所述第二目标图像和所述第一眼底图像进行鉴别，以判断所述第二目标图像是否被真实表示在所述第一图像域中。

5.根据权利要求2-4中任一项所述的方法，其特征在于，其中调整所述第一生成器、所述第二生成器、所述第一鉴别器和所述第二鉴别器的参数包括：

调整所述第一生成器、所述第二生成器、所述第一鉴别器和所述第二鉴别器的参数，使得最大化所述第一鉴别器损失函数和所述第二鉴别器损失函数以及最小化所述第一生成器损失函数和所述第二生成器损失函数。

6.根据权利要求4所述的方法，其特征在于，其中对所述第一眼底图像集和所述第二眼底图像集进行处理还包括：

将所述第一眼底图像和所述第一目标图像分别输入所述目标检测器，以分别相应地得到第一分割掩模和第二分割掩模；

基于所述第一分割掩模和所述第二分割掩模，计算第一分割损失函数；

将所述第二眼底图像和所述第二目标图像分别输入所述目标检测器，以分别相应地得到第三分割掩模和第四分割掩模；以及

基于所述第三分割掩模和所述第四分割掩模，计算第二分割损失函数。