(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202111638391.6 (22)申请日 2021.12.2 9 (71)申请人 杭州群核信息技 术有限公司 地址 310000 浙江省杭州市江干区九环路9 号3幢2楼208室 (72)发明人 陆明明　于川汇　杜明　 (74)专利代理 机构 杭州天勤知识产权代理有限 公司 33224 专利代理师 曹兆霞 (51)Int.Cl. G06T 17/10(2006.01) G06T 5/30(2006.01) G06T 7/187(2017.01) G06V 40/20(2022.01) G06V 10/44(2022.01)G06V 10/764(2022.01) G06K 9/62(2022.01) G06F 30/13(2020.01) (54)发明名称 基于图片的户型自动 构造方法、 装置和计算 机设备 (57)摘要 本发明公开了一种基于图片的户型自动构 造方法、 装置和计算机设备， 包括： 利用语义 分割 模型对户型图片进行户型元素分类； 利用关键点 检测模型提取户型元素分类结果中的候选关键 点， 并对候选关键点筛选后， 基于确定的关键点 对墙体进行打断并聚类， 以构造墙体元素； 依据 户型元素分类结果进行图像连通域检测， 以构造 门元素、 窗元素、 柱元素， 并将这三种元素与墙体 元素进行关联； 结合关联结果对构建的墙体元 素、 门元素、 窗元素以及 柱元素进行修正， 以得到 构造的户型图。 该方法可以提升户型自动构造的 准确性和效率。 权利要求书2页 说明书6页 附图4页 CN 114494636 A 2022.05.13 CN 114494636 A 1.一种基于图片的户型自动构造方法， 其特 征在于， 包括以下步骤： 利用语义分割模型对户型图片进行户型 元素分类； 利用关键点检测模型提取户型元素分类结果中的候选关键点， 并对候选关键点筛选 后， 基于确定的关键点对墙 体进行打断并聚类， 以构造墙 体元素； 依据户型元素分类结果进行图像连通域检测， 以构造门元素、 窗元素、 柱元素， 并将这 三种元素与墙体元素进行关联； 结合关联结果对构建的墙体元素、 门元素、 窗元素以及柱元素进行修正， 以得到构造的 户型图。 2.根据权利要求1所述的基于图片的户型自动构造方法， 其特征在于， 户型元素分类之 前， 对户型图片进 行归一化操作， 包括对户型图片 重置图片尺寸、 根据标准值和方差对户型 图片进行像素 标准化； 归一化操作后的户型图片输入至语义分割模型进行户型 元素分类。 候选关键点提取之前， 对户型元素分类结果进行二值化和归一化操作， 其中， 归一化操 作包括对户型元素分类结果重置图片尺寸、 根据标准值和方差对户型元素分类结果进 行像 素标准化； 归一化操作后的户型 元素分类结果输入至关键点检测模型进行候选关键点 提取。 3.根据权利要求1所述的基于图片的户型自动构造方法， 其特征在于， 所述语义分割分 模型基于P SPNet模型并经参数优化构建； 所述关键点检测模型基于 CenterNet模型并经参数优化构建。 4.根据权利要求1所述的基于图片的户型自动构造方法， 其特征在于， 依据设定阈值对 候选关键点进 行筛选以确定 关键点， 基于 关键点对户型元素分类结果中的墙体标记进 行打 断后， 对属于同一堵墙体的像素进行聚类， 并计算每个聚类簇的起始像素、 终止像素以及像 素区域的厚度， 以构造墙 体元素。 5.根据权利要求1所述的基于图片的户型自动构造方法， 其特征在于， 在构造墙体元素 之后， 提取户型元素分类结果中的承重墙区域， 判断墙体元素与承重墙区域的重合面积， 将 大于设定第一重合 面积阈值的墙 体元素作为承重墙元 素。 6.根据权利要求1所述的基于图片的户型自动构造方法， 其特征在于， 所述依据户型元 素分类结果进行图像连通 域检测， 包括： 依据户型元素分类结果， 将相同像素值的相邻像素聚类成一个像素连通区域， 并将像 素区域进行图像膨胀， 依据设置阈值对图像膨胀结果进行筛选， 将筛选得到的像素连通区 域并结合对应的像素分类， 来构建门元 素、 窗元素、 柱元素。 7.根据权利要求6所述的基于 图片的户型自动构造方法， 其特征在于， 将门元素、 窗元 素、 柱元素与墙体元素进行关联， 包括： 对每个像素连通区域进行轮廓 提取后， 计算墙体元素分别与门元素、 窗元素、 柱元素的 重叠面积， 若重 叠面积超过第二重 叠面积阈值， 则关联墙 体元素与门元 素、 窗元素、 柱元素。 8.根据权利要求1所述的基于图片的户型自动构造方法， 其特征在于， 所述结合关联结 果对构建的墙 体元素、 门元素、 窗元素以及柱元 素进行修 正， 包括： 针对墙体元素， 对墙体元素计算闭合环， 每个闭合环形成一个房间元素， 并对所有墙体 元素进行共线判断， 合并同一类型墙体， 并将共线墙体中最长墙体厚度作为合并后墙体的权　利　要　求　书 1/2 页 2 CN 114494636 A 2厚度； 针对门元素、 窗元素， 依据门元素、 窗元素与墙体元素的关联结果， 按照门、 窗附属的墙 体计算门中线、 窗中线， 并进行归一化处理， 包括： 门元素、 窗元素的长度不能超过所属墙 体， 合并共线门、 窗， 并按照最大面积的门元素、 窗元素设定合并结构的门类型、 窗类型， 去 除非法的门中中线和窗中线相交的门元素或窗元素， 并保留相交元素中面积 较大的门元素 或窗元素； 针对柱元素， 当柱元素与墙体存在交集时， 计算墙体切割柱后形成的轮廓的外接矩形 作为柱的外轮廓； 当柱元 素与墙体不存在交集时， 则计算柱元 素的外接矩形作为外轮廓。 9.一种基于图片的户型自动构造装置， 其特 征在于， 包括： 户型元素分类模块， 用于利用语义分割模型对户型图片进行户型 元素分类； 户型元素构造模块， 用于利用关键点检测模型提取户型元素分类结果中的候选关键 点， 并对候选关键点筛选后， 基于确定的关键点对墙体进行打断并聚类， 以构造墙体元素； 还用于依据户型元素分类结果进行图像连通域检测， 以构造门元素、 窗元素、 柱元素， 并将 这三种元素与墙体元素进行关联； 户型修正模块， 结合关联结果对构建的墙体元素、 门元素、 窗元素以及柱元素进行修 正， 以得到构造的户型图。 10.一种计算机设备， 包括， 存储器、 处理器以及存储在所述存储器中并可在所述处理 器上执行的计算机程序， 其特征在于， 所述处理器执行所述计算机程序时实现权利要求 1‑8 任一项所述的基于图片的户型自动构造方法的步骤。权　利　要　求　书 2/2 页 3 CN 114494636 A 3