摘要
本发明提供一种机车设计图片生成方法、装置、电子设备和存储介质,涉及机车设计技术领域,其中方法包括:获取包括随机噪声的图片和与机车设计需求相关的文本信息;将所述图片输入图像编码器中,得到所述图像编码器输出的图片潜在特征向量,并将所述文本信息输入文本编码器中,得到所述文本编码器输出的文本特征向量;将所述图片潜在特征向量和所述文本特征向量输入噪声预测模型中,得到所述噪声预测模型输出的预测噪声;基于所述预测噪声对所述图片潜在特征向量进行去噪,得到去噪后的隐式向量;对所述去噪后的隐式向量进行解码,得到机车设计图片,从而实现了机车设计图片的自动设计,提高了机车设计图片的生成效率。