基坑支撑方案生成模型的训练方法、装置、设备及介质

本申请公开了一种基坑支撑方案生成模型的训练方法、装置、设备及介质，该方法包括：获取基坑CAD图集和基坑CAD图集中每一个CAD图所对应的生成图，并对基坑CAD图集中的每一个CAD图进行预处理，得到训练数据，其中，训练数据包括基坑外轮廓图、打孔轮廓图、基坑描述信息和生成图；对训练数据进行数据扩展处理，得到每一个训练数据对应的扩展图像，其中，扩展图像为噪声图像；加载预训练好的深度学习模型，并基于扩展图像对深度学习模型进行微调处理，得到微调处理后的深度学习模型；根据训练数据和扩展图像对微调处理后的深度学习模型进行模型训练，并在训练完成时得到训练好的基坑支撑方案生成模型。提高了集成支撑方案图生成的效率和准确性。