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

说明书

本申请实施例提供一种家装设计知识图谱构建方法、装置、电子设备及存储介质，涉及家装技术领域。该方法包括获取已有设计数据的俯视图；识别所述俯视图，以获得空间数据；利用预设的目标检测算法识别出所述俯视图中的物体类别和坐标，以及物体间的位置关系；格式化并存储位置关系数据和所述空间数据，以生成所述知识图谱，获取大量已有设计方案中的空间类型、面积和形状，家具的类型、位置坐标和家具之间的位置关系，从而实现知识图谱的构建，使用范围更加广泛，解决现有方法局限定较大的问题。

技术领域

本申请涉及家装技术领域，具体而言，涉及家装设计知识图谱构建方法、装置、电子设备及存储介质。

背景技术

现有的家装知识图谱是基于爬取互联网上的一些家装知识，这些知识比较零散，没有系统性和逻辑性，并且没有对家装设计中的家具位置布置和家具之间的风格和颜色搭配做系统的知识构建，具有较大的局限性。

发明内容

本申请实施例的目的在于提供一种家装设计知识图谱构建方法、装置、电子设备及存储介质，获取大量已有设计方案中的空间类型、面积和形状，家具的类型、位置坐标和家具之间的位置关系，从而实现知识图谱的构建，使用范围更加广泛，解决现有方法局限定较大的问题。

本申请实施例提供了一种家装设计知识图谱构建方法，所述方法包括：

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

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

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

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

在上述实现过程中，由于已有的设计方案中包括了很多专业的设计理念以及各个空间中需要布置哪些家具以及家具之间的相对布置关系，家具之间的风格搭配关系，因此可利用已有的大量设计方案，获取其中的空间类型、面积和形状，家具的类型、位置坐标和家具之间的位置关系，从而实现知识图谱的构建，使用范围更加广泛，解决现有方法局限定较大的问题。

进一步地，所述识别所述俯视图，以获得空间数据，包括：

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

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

在上述实现过程中，对户型图的俯视图进行识别，从而得到空间的类别和面积数据。

进一步地，所述利用预设的目标检测算法识别出所述俯视图中的物体类别和坐标，包括：

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

在上述实现过程中，利用开源yolo-v5算法科识别出墙、门、窗和各个家具的类别以及对应的矩形框坐标。

进一步地，所述利用预设的目标检测算法识别出所述俯视图中的物体间的位置关系，包括：

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

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

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

在上述实现过程中，利用墙的矩形框坐标可计算出墙的朝向和家具的朝向以及各个物体之间的位置关系。

进一步地，所述利用所述墙的矩形框坐标计算墙的朝向，包括：

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