风格转换模型的训练方法及虚拟建筑检测模型的训练方法

摘要

本公开关于一种风格转换模型的训练方法及虚拟建筑检测模型的训练方法，属于图像处理技术领域。该风格转换模型的训练方法包括：将每个样本虚拟图像与样本真实图像输入至风格转换模型的第一生成器，以生成每个样本虚拟图像对应的转换图像，以及每个样本虚拟图像与样本真实图像的注意力热力图；将每个转换图像和样本真实图像输入至风格转换模型的判别器，以生成每个转换图像与样本真实图像的判别结果与注意力热力图；根据判别器的判别结果、各注意力热力图，分别生成对抗损失值与类激活映射损失值；根据对抗损失值与类激活映射损失值和，对风格转换模型进行训练。由此，改善了模型的虚拟图像真实化的效果，提升了虚拟图像的真实度。