[发明专利]家装设计知识图谱构建方法、装置、电子设备及存储介质

权利要求书

1.一种家装设计知识图谱构建方法，其特征在于，所述方法包括：

获取已有设计数据的俯视图；

识别所述俯视图，以获得空间数据；

利用预设的目标检测算法识别出所述俯视图中的物体类别和坐标，以及物体间的位置关系；

格式化并存储位置关系数据和所述空间数据，以生成所述知识图谱。

2.根据权利要求1所述的家装设计知识图谱构建方法，其特征在于，所述识别所述俯视图，以获得空间数据，包括：

利用目标检测算法框出所述俯视图中的空间名和面积；

利用文字识别算法对所述空间名和面积对应的文字进行识别，以获得所述空间的类别和面积。

3.根据权利要求1所述的家装设计知识图谱构建方法，其特征在于，所述利用预设的目标检测算法识别出所述俯视图中的物体类别和坐标，包括：

利用预先训练好的开源yolo-v5算法对所述俯视图进行识别，以获得墙、门、窗和各个家具的类别以及对应的矩形框坐标。

4.根据权利要求3所述的家装设计知识图谱构建方法，其特征在于，所述利用预设的目标检测算法识别出所述俯视图中的物体间的位置关系，包括：

利用所述墙的矩形框坐标计算墙的朝向；

基于墙的朝向，计算墙与家具的关系以及家具的朝向；

利用家具的朝向和所述家具的矩形框坐标计算家具间的位置关系和朝向。

5.根据权利要求4所述的家装设计知识图谱构建方法，其特征在于，所述利用所述墙的矩形框坐标计算墙的朝向，包括：

基于所述矩形框坐标，获取所述墙的四个坐标，分别为(x1,y1)、(x2,y2)、(x3,y3)、(x4,y4)；

获取所述墙的两端的端坐标，所述端坐标表示为[(x1,y1),(x2,y2)]和[(x3,y3),(x4,y4)]；

基于所述端坐标计算所述墙的中心线坐标[(a1,b1),(a2,b2)]；

其中，a1＝(x1+x2)/2,b1＝(y1+y2)/2；

a2＝(x3+x4)/2,b2＝(y3+y4)/2；

计算各个端坐标的最小距离，并获得首尾相连墙对应的坐标，以生成一个封闭的墙多边形；

获取所述墙多边形的最小外接矩形的中心点坐标(x,y)；

比较中心点坐标值和中心线坐标值的大小，以确定所述墙的朝向。

6.根据权利要求5所述的家装设计知识图谱构建方法，其特征在于，所述比较中心点坐标值和中心线坐标值的大小，以确定所述墙的朝向，包括：

若a1＝a2，则所述墙垂直；

若a1x，则所述墙的朝向为朝右；

若a1x，则所述墙朝左；

若b1＝b2，则所述墙水平；

若b1y，则所述墙朝上；

若b1y时，则所述墙朝下。

7.一种家装设计知识图谱构建装置，其特征在于，所述装置包括：

数据获取模块，用于获取已有设计数据的俯视图；

第一识别模块，用于识别所述俯视图，以获得空间数据；

第二识别模块，用于利用预设的目标检测算法识别出所述俯视图中的物体类别和坐标，以及物体间的位置关系；

存储模块，用于格式化并存储位置关系数据和所述空间数据，以生成所述知识图谱。

8.根据权利要求7所述的家装设计知识图谱构建装置，其特征在于，所述第一识别模块包括：

空间识别模块，用于利用目标检测算法框出所述俯视图中的空间名和面积；

文字识别模块，用于利用文字识别算法对所述空间名和面积对应的文字进行识别，以获得所述空间的类别和面积。

9.一种电子设备，其特征在于，所述电子设备包括存储器以及处理器，所述存储器用于存储计算机程序，所述处理器运行计算机程序以使计算机设备执行根据权利要求1至6中任一项所述的家装设计知识图谱构建方法。

10.一种可读存储介质，其特征在于，所述可读存储介质中存储有计算机程序指令，所述计算机程序指令被一处理器读取并运行时，执行权利要求1至6任一项所述的家装设计知识图谱构建方法。