一种轨道交通数据增强方法及系统
摘要:
本说明书实施例提供了一种轨道交通数据增强方法及系统,其中,方法包括:获取轨道交通极端场景下的数据并作为生成对抗网络模型数据集;通过所述生成对抗网络模型数据集构建基于WGAN‑GP的生成对抗网络模型,根据所述生成对抗网络模型扩充所述生成对抗网络模型数据集,其中,所述基于WGAN‑GP的生成对抗网络模型具体包括:生成器和判别器,所述生成器与所述判别器相连,所述生成器的输出作为所述判别器的输入,所述判别器的输出通过后向传播反馈给所述生成器,所述生成器包含5个依次连接的下采样模块和5个依次连接的上采样模块并对应连接,所述判别器为根据Unet网络的思想在原始判别器右边增加一个上采样路径形成的新U型判别器。
0/0