[发明专利]提取图像的特征图的方法和装置

说明书

本公开的实施例公开了提取图像的特征图的方法和装置。该方法的一具体实施方式包括：获取待检测图像；将待检测图像输入特征提取网络，特征提取网络包括至少三个特征提取层；从至少三个特征提取层输出的特征图中选取三个特征图，基于三个特征图，得到待检测图像对应的特征图。该实施方式实现了高、中、低分辨率的特征信息充分融合。

技术领域

本公开的实施例涉及计算机技术领域，具体涉及提取图像的特征图的方法和装置。

背景技术

在对图像进行处理时，可以先将图像输入特征提取网络。然后，根据得到的特征图对于图像进行多种预测。例如，姿态检测、图像分割等等。在此过程中，特征提取网络输出的特征图可以保留较多的上下文信息。但随着网络层数的加深，特征图的分辨率较低，往往损失了很多细节信息。

发明内容

本公开的实施例提出了提取图像的特征图方法和装置。

第一方面，本公开的实施例提供了一种提取图像的特征图的方法，包括：获取待检测图像；将待检测图像输入特征提取网络，特征提取网络包括至少三个特征提取层；从至少三个特征提取层输出的特征图中选取三个特征图，基于三个特征图，得到待检测图像对应的特征图。

在一些实施例中，将待检测图像的特征图输入预先训练的姿态预测网络，得到待检测图像中显示的人体的姿态信息，其中，姿态预测网络用于表征图像的特征图与图像中显示的人体的姿态信息的对应关系。

在一些实施例中，将待检测图像的特征图输入预先训练的图像分割网络，得到对于待检测图像中显示的至少一个对象的分割结果信息，其中，图像分割网络用于表征图像的特征图与对图像中显示的至少一个对象的分割结果信息的对应关系。

在一些实施例中，从至少三个特征提取层输出的特征图中选取三个特征图，基于三个特征图，得到待检测图像对应的特征图，包括：从至少三个特征提取层输出的特征图中选取分辨率满足预设条件的三个特征图，将三个特征图进行融合，得到待检测图像对应的特征图。

在一些实施例中，从至少三个特征提取层输出的特征图中选取分辨率满足预设条件的三个特征图，包括：对于至少三个特征提取层输出的特征图，按照分辨率由小到大的顺序，选取两个特征图作为第一特征图和第二特征图；选取分辨率与所述待检测图像的分辨率的比值等于预设阈值的特征图作为第三特征图。

在一些实施例中，从至少三个特征提取层输出的特征图中选取分辨率满足预设条件的三个特征图，包括：对于至少三个特征提取层输出的特征图，选取分辨率位于第一分辨率区间的特征图作为第一特征图；选取分辨率位于第二分辨率区间的特征图作为第二特征图；选取分辨率位于第三分辨率区间的特征图作为第三特征图。

第二方面，本公开的实施例提供了一种提取图像的特征图的装置，包括：获取单元，被配置成获取待检测图像；特征提取单元，被配置成将待检测图像输入特征提取网络，特征提取网络包括至少三个特征提取层；特征融合单元，被配置成从至少三个特征提取层输出的特征图中选取三个特征图，基于三个特征图，得到待检测图像对应的特征图。

在一些实施例中，该装置还包括：姿态预测单元，被配置成将待检测图像的特征图输入预先训练的姿态预测网络，得到待检测图像中显示的人体的姿态信息，其中，姿态预测网络用于表征图像的特征图与图像中显示的人体的姿态信息的对应关系。

在一些实施例中，该装置还包括：分割单元，被配置成将待检测图像的特征图输入预先训练的图像分割网络，得到对于待检测图像中显示的至少一个对象的分割结果信息，其中，图像分割网络用于表征图像的特征图与对图像中显示的至少一个对象的分割结果信息的对应关系。

在一些实施例中，特征融合单元进一步被配置成：从至少三个特征提取层输出的特征图中选取分辨率满足预设条件的三个特征图，将三个特征图进行融合，得到待检测图像对应的特征图。