基于样式生成对抗网络的红外图像中舰船样本扩充方法

作者：吴鑫; 汪钰; 邹俊锋; 李俊儒; 黄曦

来源：2020-07-08, 中国, ZL202010650897.8.

本发明提供了一种基于样式生成对抗网络的红外图像中舰船样本扩充方法。主要解决现有技术生成红外图像由于仿真建模复杂导致真实性较差,可见光-红外图像光电转换训练样本获取难度大、训练集数量少导致扩充的红外图像缺乏多样性的问题,本发明的步骤如下：(1)选取实拍红外图像组成训练集；(2)构建生成器网络；(3)构建判别器网络；(4)构建样式生成对抗网络；(5)训练判别器网络；(6)训练生成器网络；(7)训练样式生成对抗网络；(8)使用训练好的生成器网络输出红外图像样本,完成红外图像样本扩充。本发明能够大量生成红外舰船样本,有效地提高了扩充样本的真实感和多样性。

• 单位
  西安电子科技大学