基于扩散模型与物理模型的轨道交通雾霾图像生成方法

本发明提供基于扩散模型与物理模型的轨道交通雾霾图像生成方法，包括：获取轨道交通场景晴天清晰图像，通过多模态表征生成模块，生成含噪声潜向量、边缘特征图、深度特征图、语义特征图及内容提示；构建融合提示编码器，生成雾霾风格融合提示；采用分层注入式三重控制网络，将多模态特征与融合提示注入扩散模型，完成晴天到雾霾天气的风格迁移，输出雾霾风格中间图像；基于改进大气散射物理模型，结合深度与语义信息，生成密度可控、物理真实的轨道交通雾霾图像。本发明解决轨道交通雾霾样本稀缺、生成可控性差、真实度不足的问题，为轨道交通场景智能检测提供高质量数据支撑。