[发明专利]基于目标检测的户型图墙体处理方法、系统、介质及设备

权利要求书

1.一种基于目标检测的户型图墙体处理方法，其特征在于，包括：

检测户型图中家具并对所述家具所在区域进行定位；

对所述家具所在区域像素RGB进行赋值处理；

对户型图构件进行轮廓检测以获取户型图轮廓信息；

根据筛选条件过滤所述户型图轮廓获取所述户型图墙体轮廓；

填充墙体轮廓以获取标准训练户型图。

2.根据权利要求1所述的方法，其特征在于，还包括：

对所述户型图中数据集中的家具大小进行聚类统计；

通过聚类统计获取N种不同大小的检测框，对每种尺寸框定义M种不同比例的检测框，N、M为正整数；

在所述户型图中预测K个检测框并对检测框内的家具进行分类；

计算两个同类检测框的IOU，若大于预设阈值则将两个框整合成一个，直到任意两个检测框之间的IOU小于预设阈值以获取最终检测框；

定义所述最终检测框为家具所在区域并检测所述去检测家具所在区域周围竖直、水平的直线。

3.根据权利要求1或2所述的方法，其特征在于，还包括：

统计获取户型图内家具所在区域外的具有相同RGB值的像素点的占比；

比对获取占比最大的相同像素点的RGB值；

将所述RGB值替换家具所在区域内的像素点的RGB值。

4.根据权利要求3所述的方法，其特征在于，还包括：

通过cv2.findContours计算出所述户型图的轮廓并以树状图的形式保存；

对所述轮廓记录轮廓下标，轮廓点的坐标以及父轮廓的下标；

定义最外层父轮廓的下标为-1；

循环更新所述轮廓的父轮廓，记录节点的深度，直到最新节点的父轮廓下标为-1；

返回节点的深度。

5.根据权利要求4所述的方法，其特征在于，还包括：

步骤1：将所述轮廓中相邻的3个轮廓点的首尾两个轮廓点作为端点连接两点形成线段记作AB；

步骤2：计算所述3个轮廓点中中间轮廓点距离线段AB的垂直距离d；

步骤3：比较d与预设阈值threshold的大小，若d小于threshold，则以线段AB作为所述轮廓的近似构成线段；若d大于threshold，则将C作为端点形成线段AC和BC作为所述轮廓的近似构成线段；

步骤4：重复步骤1～步骤3获取所述轮廓的近似构成线段作为所述轮廓的构成线段并记录构成所述轮廓线段端点的个数。

6.根据权利要求5所述的方法，其特征在于，所述筛选条件为：

判断所述轮廓深度是否小于P，若否，则过滤所述轮廓；

计算所述轮廓的最大内接圆半径，判断所述最大内接圆半径是否在预设范围内，所否，则过滤所述轮廓；

计算获取轮廓的面积占所述户型图面积的占比，判断所述占比是否在预设范围内，若否，则过滤所述轮廓；

计算所述轮廓的周长与轮廓面积比值，判断所述比值是否大于预设值，若否，则过滤所述轮廓；

计算所述轮廓构成线段的端点数，判断所述端点数是否在预设范围内，若否，则过滤所述轮廓。

7.根据权利要求1或6所述的方法，其特征在于，

获取构成所述户型图轮廓像素点的RGB值；

将所述RGB值替换所述户型图墙体轮廓内的像素点的RGB值。

8.一种基于目标检测的户型图墙体处理系统，其特征在于，所述系统包括：

定位模块，用于检测户型图中家具并对所述家具所在区域进行定位；

检测模块，用于对户型图构件进行轮廓检测以获取户型图轮廓信息；

过滤模块，用于根据筛选条件过滤所述户型图轮廓获取所述户型图墙体轮廓；

填充模块，用于对所述家具所在区域和所述墙体轮廓进行像素填充。

9.一种计算机可读存储介质，其中存储有计算机程序，其特征在于，所述计算机程序被处理器加载执行时，实现如权利要求1至7中任一所述的基于目标检测的户型图墙体处理方法。

10.一种电子设备，其特征在于，包括：处理器、存储器及通信接口；其中，

所述存储器用于存储计算机程序；

所述处理器用于加载执行所述计算机程序，以使所述电子设备执行如权利要求1至7中任一所述的基于目标检测的户型图墙体处理方法；

所述通信接口用于实现访问装置与其他设备之间的通信。