[发明专利]图像区域处理方法及设备

权利要求书

1.一种图像区域处理方法，包括：

获取目标图像和终端在拍摄所述目标图像时的设备姿态；

对所述目标图像进行图像区域识别，得到初始识别结果，所述图像区域包括天花板区域、墙壁区域和地面区域中的至少一种；

根据所述设备姿态对所述初始识别结果进行修正，得到修正后的识别结果。

2.根据权利要求1所述的图像区域处理方法，所述设备姿态包括所述终端的仰角和/或所述终端的俯角，所述根据所述设备姿态对所述初始识别结果进行修正，得到修正后的识别结果，包括：

将所述终端的仰角和/或所述终端的俯角与角度阈值进行比较；

根据比较结果对所述初始识别结果进行修正，得到所述修正后的识别结果。

3.根据权利要求2所述的图像区域处理方法，所述根据比较结果对所述初始识别结果进行修正，得到所述修正后的识别结果，包括：

如果所述比较结果为所述终端的仰角大于第一阈值，则将所述初始识别结果中的地面区域重新识别为天花板区域，得到所述修正后的识别结果。

4.根据权利要求2所述的图像区域处理方法，所述根据比较结果对所述初始识别结果进行修正，得到所述修正后的识别结果，包括：

如果所述比较结果为所述终端的俯角大于第二阈值，则将所述初始识别结果中的天花板区域重新识别为地面区域，得到所述修正后的识别结果。

5.根据权利要求3所述的图像区域处理方法，所述根据比较结果对所述初始识别结果进行修正，得到所述修正后的识别结果，包括：

如果所述比较结果为所述终端的仰角大于第三阈值，则将所述初始识别结果中的地面区域、墙壁区域重新识别为天花板区域，得到所述修正后的识别结果，其中，所述第三阈值大于所述第一阈值。

6.根据权利要求4所述的图像区域处理方法，所述根据比较结果对所述初始识别结果进行修正，得到所述修正后的识别结果，包括：

如果所述比较结果为所述终端的俯角大于第四阈值，则将所述初始识别结果中的墙壁区域、天花板区域重新识别为地面区域，得到所述修正后的识别结果，其中，所述第四阈值大于所述第二阈值。

7.根据权利要求2至6中任一项所述的图像区域处理方法，所述初始识别结果包括所述目标图像的多个像素点对应的区域概率，所述区域概率包括天花板概率、墙壁概率、地面概率中的至少一种，所述根据比较结果对所述初始识别结果进行修正，得到所述修正后的识别结果，包括：

根据所述比较结果，在所述初始识别结果中，对所述目标图像的多个像素点对应的区域概率进行调整；

根据所述目标图像的多个像素点对应的调整后的区域概率，得到所述修正后的识别结果。

8.根据权利要求1至6中任一项所述的图像区域处理方法，所述对所述目标图像进行图像区域识别，得到初始识别结果，包括：

通过图像识别模型对所述目标图像进行图像区域识别，得到所述初始识别结果，所述图像识别模型为通过模型蒸馏的方式训练得到的深度学习模型。

9.根据权利要求1至6中任一项所述的图像区域处理方法，所述得到修正后的识别结果后，还包括：

在所述目标图像上，显示所述修正后的识别结果。

10.一种图像区域处理设备，包括：

获取单元，用于获取目标图像和终端在拍摄所述目标图像时的设备姿态；

识别单元，用于对所述目标图像进行图像区域识别，得到初始识别结果，所述图像区域包括天花板区域、墙壁区域和地面区域中的至少一种；

修正单元，用于根据所述设备姿态对所述初始识别结果进行修正，得到修正后的识别结果。

11.一种电子设备，包括：至少一个处理器和存储器；

所述存储器存储计算机执行指令；

所述至少一个处理器执行所述存储器存储的计算机执行指令，使得所述至少一个处理器执行如权利要求1至9任一项所述的图像区域处理方法。