建筑图纸构件分类方法、装置、电子设备及存储介质

摘要

本申请实施例公开了一种建筑图纸构件分类方法、装置、电子设备及存储介质，该方法包括：对输入的训练数据集中的建筑图纸进行预处理，得到目标建筑图纸；然后通过全卷积分割网络对目标建筑图纸进行下采样得到第一特征图，对第一特征图进行上采样得到第二特征图；采用阈值分割法对第二特征图中每个像素点进行类别判断，输出建筑图纸中每个构件的分类结果；重复以上步骤训练出针对单个构件进行分类的第一分类器和针对所有构件进行统一分类的第二分类器；根据第一分类器和第二分类器在测试数据集上的测试效果确定最终要使用的目标分类器，使用目标分类器对待分类建筑图纸中的构件进行分类。本申请实施例有利于提高建筑图纸构件分类的效果。