基于YOLO模型的空间识别方法、系统、终端及介质

本发明公开了基于YOLO模型的空间识别方法、系统、终端及介质，方法包括：获取CAD图纸，并对CAD图纸进行空间分割与坐标转换，得到CAD图纸中指定空间区域对应的PNG图像以及PNG图像所对应的标签文件，基于PNG图像与标签文件，确定训练数据集；搭建Python运行环境，安装YOLOv8‑seg模型及其相关依赖库，并基于YOLOv8‑seg模型的训练脚本对训练数据集进行端到端训练，得到YOLO空间分割模型；基于YOLO空间分割模型对建筑平面图纸进行空间分割，输出分割结果，并进行空间区域的自动化标注。本发明采用YOLO空间分割模型，实现高精度的空间识别与分类，泛化能力强。