[发明专利]主体识别方法和装置、电子设备、计算机可读存储介质

权利要求书

1.一种主体识别方法，其特征在于，包括：

获取原始图像；

将所述原始图像作为待识别图像，对所述待识别图像进行主体识别，得到主体区域；

当所述主体区域不满足预设要求时，基于所述主体区域，从所述原始图像中获取迭代图像；所述迭代图像包含所述主体区域；

将所述迭代图像作为待识别图像，迭代执行所述对所述待识别图像进行主体识别，得到主体区域步骤；

当所述主体区域满足预设要求，将满足预设要求的所述主体区域作为目标主体区域；

所述对所述待识别图像进行主体识别，得到主体区域，包括：

生成与所述待识别图像对应的中心权重图，其中，所述中心权重图所表示的权重值从中心到边缘逐渐减小；

将所述待识别图像和所述中心权重图输入到主体识别模型中，得到主体区域置信度图，其中，所述主体识别模型是预先根据同一场景的待识别图像、中心权重图及对应的已标注的主体掩膜图进行训练得到的模型；

根据所述主体区域置信度图确定所述待识别图像中的主体区域。

2.根据权利要求1所述的方法，其特征在于，所述当所述主体区域不满足预设要求时，基于所述主体区域，从所述原始图像中获取迭代图像，包括：

当所述主体区域不满足预设要求时，基于所述主体区域，获取扩张范围；

根据所述扩张范围从所述原始图像中获取迭代图像。

3.根据权利要求2所述的方法，其特征在于，所述当所述主体区域不满足预设要求时，基于所述主体区域，获取扩张范围，包括：

当所述主体区域不满足预设要求时，确定所述主体区域的面积占所述原始图像的面积的比例；

当所述比例大于比例阈值时，获取第一扩张范围；

所述根据所述扩张范围从所述原始图像中获取迭代图像，包括；

根据所述第一扩张范围从所述原始图像中获取迭代图像。

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

当所述比例小于或等于比例阈值时，获取第二扩张范围；所述第二扩张范围大于所述第一扩张范围；

所述根据所述扩张范围从所述原始图像中获取迭代图像，包括；

根据所述第二扩张范围从所述原始图像中获取迭代图像。

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

对所述待识别图像进行主体识别，得到至少两个候选区域；

确定至少两个所述候选区域的位置信息；

根据各个所述候选区域的位置信息从所述至少两个候选区域中确定主体区域。

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

连续的预设数量的主体区域的差异度小于差异度阈值。

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

当所述迭代次数达到数量阈值时，将最后一次迭代得到的主体区域作为目标主体区域。

8.一种主体识别装置，其特征在于，包括：

原始图像获取模块，用于获取原始图像；

主体识别模块，用于将所述原始图像作为待识别图像，对所述待识别图像进行主体识别，得到主体区域；

迭代图像获取模块，用于当所述主体区域不满足预设要求时，基于所述主体区域，从所述原始图像中获取迭代图像；所述迭代图像包含所述主体区域；

迭代模块，用于将所述迭代图像作为待识别图像，迭代执行所述对所述待识别图像进行主体识别，得到主体区域步骤；

目标主体区域获取模块，用于当所述主体区域满足预设要求，将满足预设要求的所述主体区域作为目标主体区域；

所述主体识别模块还用于生成与所述待识别图像对应的中心权重图，其中，所述中心权重图所表示的权重值从中心到边缘逐渐减小；将所述待识别图像和所述中心权重图输入到主体识别模型中，得到主体区域置信度图，其中，所述主体识别模型是预先根据同一场景的待识别图像、中心权重图及对应的已标注的主体掩膜图进行训练得到的模型；根据所述主体区域置信度图确定所述待识别图像中的主体区域。