基于手绘草图的CAD草图生成方法及系统

本发明提出一种基于手绘草图的CAD草图生成方法及系统，属于计算机辅助设计领域，包括：S1：构建CAD草图数据集；S2：基于Transformer模型进行改进得到CAD草图生成模型，将CAD草图数据集中手绘化图像输入CAD草图生成模型，输出预测的基元类型、基元布尔标志、基元参数以及约束关系矩阵；S3：构建总体损失函数训练CAD草图生成模型，得到训练好的模型；S4：将手绘草图输入训练好的的CAD草图生成模型，生成CAD草图的图结构；S5：对图结构中基元和约束进行渲染和可视化。本发明方法加速模型处理图像的速度，无需引入额外的约束查询，计算量更小。