一种基于扩散模型的矢量图纸生成模型训练方法及系统

本发明公开了一种基于扩散模型的矢量图纸生成模型训练方法及系统，方法包括：对原始图像进行预处理后，得到目标图像，并将目标图像与编辑指令输入扩散模型中，基于扩散模型进行正向扩散，得到带噪声的空间图像；基于编辑指令、原始图像以及带噪声的空间图像构建条件输入，并基于条件输入确定扩散模型的损失函数；通过反向传播算法优化扩散模型的模型参数，并使用优化器更新扩散模型中的权重，并基于循环训练与验证的方式，得到矢量图纸生成模型。本发明可较为高效、准确地、灵活地生成满足审查方和施工方要求的尺寸标注图像，提高设计质量和效率。