[发明专利]基于GAN的加油站危险场景图片生成方法

权利要求书

1.一种基于GAN的加油站危险场景图片生成方法，其特征在于，所述方法包括步骤：

图片输入单元输入危险场景图片以及加油站背景图片；

图片生成单元根据所述输入的图片及生成的图片生成新的加油站危险场景图片；

图片判决单元判决生成的危险场景图片是否真实；

若图片判决单元判决结果为不真实，由图片生成单元继续迭代生成新的图片数据；

若图片判决单元判决结果为真实，输出有效的加油站危险场景图片训练样本数据；

所述GAN为生成对抗网络（Generative Adversarial Network, GAN）。

2.如权利要求1所述的一种基于GAN的加油站危险场景图片生成方法，其特征在于，所述图片输入单元输入危险场景图片以及加油站背景图片，具体包括：

所述图片输入单元输入图片的方式可以是通过网络传输，也可以通过存储设备；

所述危险场景图片包括但不限于：抽烟场景、打手机场景、烟雾场景、火焰场景的图片数据；

所述危险场景图片数据不是在加油站场景进行采集的；

所述加油站背景图片是指各种环境条件下加油站的背景图片；

所述图片输入单元将采集到的危险场景图片数据集、加油站背景图片数据集以及标注信息作为原始数据源输入到图片生成单元。

3.如权利要求1所述的一种基于GAN的加油站危险场景图片生成方法，其特征在于，所述图片生成单元根据所述输入的图片及生成的图片生成新的加油站危险场景图片，具体包括：

所述输入的图片是指所述图片输入单元输入的危险场景图片数据集、加油站背景图片数据集以及标注信息的原始数据源；

所述生成的图片是指经过生成的有效的加油站危险场景图片数据集；

所述图片生成单元基于所述危险场景图片数据集生成新的危险场景图片数据；

所述图片生成单元基于所述加油站背景图片数据集生成新的加油站背景图片数据；

所述图片生成单元还用于将所述新的危险场景图片数据和所述新的加油站背景图片数据进行合成，生成加油站危险场景图片数据；

所述加油站危险场景图片数据包括但不限于：加油站抽烟场景、加油站打手机场景、加油站烟雾场景、加油站火焰场景的图片数据。

4.如权利要求1所述的一种基于GAN的加油站危险场景图片生成方法，其特征在于，所述图片判决单元判决生成的危险场景图片是否真实，具体包括：

所述图片判决单元用于判决生成的加油站抽烟场景的图片是否是真实的加油站抽烟场景；

所述图片判决单元用于判决生成的加油站打手机场景的图片是否是真实的加油站打手机场景；

所述图片判决单元用于判决生成的加油站烟雾场景的图片是否是真实的加油站烟雾场景；

所述图片判决单元用于判决生成的加油站火焰场景的图片是否是真实的加油站火焰场景。

5.如权利要求1所述的一种基于GAN的加油站危险场景图片生成方法，其特征在于，所述若图片判决单元判决结果为不真实，由图片生成单元继续迭代生成新的图片数据，具体包括：

若图片判决单元判决所述生成的加油站抽烟场景的图片的结果为不真实，则由图片生成单元继续迭代生成新的加油站抽烟场景的图片数据；

若图片判决单元判决所述生成的加油站打手机场景的图片的结果为不真实，则由图片生成单元继续迭代生成新的加油站打手机场景的图片数据；

若图片判决单元判决所述生成的加油站烟雾场景的图片的结果为不真实，则由图片生成单元继续迭代生成新的加油站烟雾场景的图片数据；

若图片判决单元判决所述生成的加油站火焰场景的图片的结果为不真实，则由图片生成单元继续迭代生成新的加油站火焰场景的图片数据。