基于风格迁移的航空摄影图像目标样本生成方法

摘要

本发明公开一种基于风格迁移的航空摄影图像目标样本生成方法，包括以下步骤：S1：在航空摄影图像中，选取具有代表性的多种地物场景的图像并裁剪；S2：随机选取S1中的航空摄影图像，将其设置为虚拟场景的背景图像；S3：对三维模型进行预处理；S4：将三维模型导入虚拟场景中；S5：输出合成样本图像；S6：输出标签文件；S7：构建风格迁移算法的训练数据集；S8：构造循环一致生成对抗网络；S9：利用训练完成的风格迁移网络模型，推理得到逼真合成样本图像，并将S6中得到的标签数据与逼真合成样本图像对应，构成逼真合成样本。本发明为航空摄影图像目标检测的合成样本图像的生成和标注提供了一种高效的手段。