建筑门窗CAD检测数据集 深度学习YOLO模型如何训练CAD图纸检测数据集

建筑门窗CAD检测数据集 深度学习YOLO模型如何训练CAD图纸检测数据集

建筑门窗CAD检测数据集

数据集信息表

项目详细说明
数据集名称建筑门窗检测数据集
图片总量2587张(约2.6k)
标注类别(共3类)门、窗户、区(区域/构件)
数据集版本2个版本
标注格式支持YOLO、XML、JSON等主流格式
适用场景建筑图纸/实景门窗识别、房屋构件检测、AI目标检测模型训练


1

YOLO训练配置文件(building.yaml

path:./building_datasettrain:images/trainval:images/valtest:images/testnc:3names:0:door1:window2:area

最简训练代码

fromultralyticsimportYOLOdefmain():model=YOLO("yolov8s.pt")model.train(data="./building_dataset/building.yaml",epochs=100,imgsz=640,batch=16,device=0,project="runs/train",name="door_window_detect")if__name__=="__main__":main()