[发明专利]临摹图识别方法、装置、终端设备及存储介质

权利要求书

1.一种临摹图识别方法，其特征在于，包括：

从获取的临摹图中提取房间类型特征图和房间边界特征图；

根据所述房间类型特征图确定注意力权重；

根据所述房间类型特征图和所述注意力权重，对所述房间边界特征图进行融合，以输出房间边界的分类结果。

2.如权利要求1所述的临摹图识别方法，其特征在于，所述根据所述房间类型特征图获取注意力权重包括：

对所述房间类型特征图进行特征映射，以获取房间子特征图；

将所述房间子特征图的权重作为第一注意力权重；

将所述第一注意力权重和预设的方向感知卷积核进行聚合，以得到第二注意力权重。

3.如权利要求2所述的临摹图识别方法，其特征在于，所述根据所述房间类型特征图和所述注意力权重，对所述房间边界特征图进行融合，以输出房间边界的分类结果包括：

获取所述房间边界特征图的边界子特征图；

对边界子特征图与所述第二注意力权重执行第一聚合，得到第一边界图；

对所述第一边界图与第一注意力权重执行第二聚合，得到第二边界图；

根据所述房间边界特征图和所述第二边界图，确定房间边界分类结果。

4.如权利要求3所述的临摹图识别方法，其特征在于，所述根据所述房间边界特征图和所述第二边界图，确定房间边界分类结果包括：

对所述房间边界特征图和所述第二边界图进行串联，得到边界分类图；

将所述边界分类图输入到预设的分类网络中，并输出该边界分类图对应的分类概率；

将最大分类概率对应的边界结果确定为房间边界分类结果。

5.如权利要求1-4任意一项所述的临摹图识别方法，其特征在于，所述从获取的临摹图中提取房间类型特征图和房间边界特征图包括：

对所述临摹图进行卷积池化处理，以获取第一共享特征图；

将所述第一共享特征图经过空洞卷积处理，以得到第二共享特征图；

对所述第二共享特征图进行解码，以输出房间类型特征图和房间边界特征图。

6.如权利要求1-4任意一项所述的临摹图识别方法，其特征在于，所述从获取的临摹图中提取房间类型特征图和房间边界特征图包括：

对所述临摹图进行卷积池化处理，以获取第一共享特征图；

通过预设的残差网络从所述第一共享特征图中提取第二共享特征图；

对所述第二共享特征图进行解码，以输出房间类型特征图和房间边界特征图。

7.一种临摹图识别装置，其特征在于，包括：

获取模块，用于从获取的临摹图中提取房间类型特征图和房间边界特征图；

注意力模块，用于根据所述房间类型特征图确定注意力权重；

分类模块，用于根据所述房间类型特征图和所述注意力权重，对所述房间边界特征图进行融合，以输出房间边界的分类结果。

8.如权利要求7所述的临摹图识别装置，其特征在于，所述注意力模块包括：

映射单元，用于对所述房间类型特征图进行特征映射，以获取房间子特征图；

第一注意力单元，用于将所述房间子特征图的权重作为第一注意力权重；

第二注意力单元，用于将所述第一注意力权重和预设的方向感知卷积核进行聚合，以得到第二注意力权重。

9.一种终端设备，包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时实现如权利要求1至6任一项所述临摹图识别方法的步骤。

10.一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至6任一项所述临摹图识别方法的步骤。