一种海上场景图像生成方法及装置

本申请提供一种海上场景图像生成方法及装置，方法包括：确定海上场景的真实图像为期望生成目标的背景图像；生成目标的目标类型，在背景图像中期望生成目标的目标位置创建掩膜图像，掩膜图像与目标类型相匹配；将掩膜图像与背景图像共同输入至扩散模型，扩散模型的编码器将图像信息转换成低维的潜在表示；在隐空间中对潜在表示进行正扩散与逆扩散，正扩散过程通过加噪处理得到潜在表示在任意时刻的状态，逆扩散过程基于U‑Net网络进行去噪处理，通过噪声预测得到目标图像潜在表示；基于目标图像潜在表示生成海上场景的目标图像。本方案生成虚实结合的海上目标训练样本，提高了生成样本的质量和多样性，降低了数据获取成本。