一种基于Pix2Pix的矢量图纸生成模型训练方法

本发明公开了一种基于Pix2Pix的矢量图纸生成模型训练方法，方法包括：获取若干真实建筑平面图，构建配对图集，配对图集中反映的是原始建筑平面图与目标建筑平面图之间的对应关系；将配对图集输入至生成对抗网络模型中进行对抗训练，输出对抗训练的反馈结果；基于反馈结果，对生成对抗网络模型进行迭代优化，直至生成对抗网络模型的对抗训练收敛，得到矢量图纸生成模型。本发明基于配对图集训练生成对抗网络模型，从而生成一个可以自动根据输入的原始建筑平面图输出目标建筑平面图的矢量图纸生成模型，便于实现对CAD图纸的智能生成、标注、修复、风格迁移等功能，提高CAD图纸处理的效率与质量，降低人工干预成本。