一种用于SWE图像生成的生成对抗模型构建方法及装置

本发明涉及一种用于SWE图像生成的生成对抗模型构建方法与装置，方法包括：对获取的目标图像数据集进行预处理并划分为训练集、验证集、测试集；构建生成对抗模型，所述生成对抗模型包括生成器和鉴别器；将所述BUMS图像作为生成器的输入，生成器输出生成SWE图像，提取真实SWE图像和生成SWE图像各自对应的RGB三通道图像，结合所述RGB三通道图像，计算生成SWE图像与真实SWE图像间的损失值，并根据损失值反向传播调整生成器参数，同时将生成SWE图像和真实SWE图像作为鉴别器的输入，计算鉴别器对于输入图像的鉴别损失，并根据鉴别损失反向传播调整鉴别器参数；利用所述测试集测试所述生成对抗模型，并根据测试结果评价所述生成对抗模型。