[发明专利]图像背景特征提取方法、装置、设备及介质

权利要求书

1.一种图像背景特征提取方法，其特征在于，所述方法包括：

利用预设分割模型对待处理图像进行分割，得到所述待处理图像的人像掩膜，合并所述人像掩膜与所述待处理图像得到人像分割图；

对所述人像分割图进行填充得到填充图像，分割所述填充图像得到n x n个小区块的人像填充分割图，将所述n x n个小区块展开成1xn²的非背景掩膜，所述人像填充分割图包含有背景区域、非背景区域；

对所述非背景掩膜进行矩阵计算得到非背景自注意力掩膜，根据所述非背景自注意力掩膜对应的掩膜矩阵和所述人像填充分割图对应的分割图矩阵生成自注意力矩阵，根据所述非背景自注意力掩膜，对所述背景区域的小区块两两为一组标记关注标签；

将所述自注意力矩阵和所述分割图矩阵输入预设自注意力模型进行矩阵相乘，根据所述关注标签，从所述矩阵相乘的结果中提取所有数值为1的元素在所述人像填充分割图中对应的小区块的特征，作为所述待处理图像的背景特征。

2.如权利要求1所述的图像背景特征提取方法，其特征在于，所述利用预设分割模型对待处理图像进行分割，得到所述待处理图像的人像掩膜，包括：

对所述待处理图像进行特征提取，得到所述待处理图像的特征图；

根据预设人像区域取选框对所述特征图的人像部分进行分割，得到所述人像掩膜。

3.如权利要求2所述的图像背景特征提取方法，其特征在于，所述根据预设人像区域取选框对所述特征图的人像部分进行分割，得到所述人像掩膜，包括：

对所述特征图进行上采样处理，扩充所述特征图至预设分辨率得到扩充特征图；

根据所述人像区域取选框，区分所述扩充特征图的每一个像素得到所述特征图的人像部分并进行分割，根据分割的人像部分生成所述待处理图像的人像掩膜。

4.如权利要求1所述的图像背景特征提取方法，其特征在于，所述对所述人像分割图进行填充得到填充图像，包括：

读取所述人像分割图的各边长的数值，选取数值最大的一条边；

以所述选取的边对所述人像分割图参照预设正方框按比例进行缩放，以使所述选取的边的边长等于所述预设正方框的边长；

对所述人像分割图未超过预设正方框的区域进行图像填充，得到所述填充图像。

5.如权利要求1所述的图像背景特征提取方法，其特征在于，所述对所述非背景掩膜进行矩阵计算得到非背景自注意力掩膜，根据所述非背景自注意力掩膜对应的掩膜矩阵和所述人像填充分割图对应的分割图矩阵生成自注意力矩阵，包括：

提取所述非背景掩膜的每个小区块的特征进行矩阵计算得到非背景注意力掩膜；

及提取所述人像填充分割图的每个小区块的特征生成所述分割图矩阵；

读取所述掩膜矩阵的每个元素和所述分割图矩阵的每个元素进行矩阵相乘生成所述自注意力矩阵。

6.如权利要求1所述的图像背景特征提取方法，其特征在于，所述根据所述非背景自注意力掩膜，对所述背景区域的小区块两两为一组标记关注标签，包括：

根据所述非背景自注意力掩膜，对所述人像填充分割图的每个小区块之间的关联性进行判断；

若判断两两为一组的小区块的数值同时大于或等于预设值，则将所述两两为一组的小区块作为背景区域并标记为关注标签。

7.如权利要求6所述的图像背景特征提取方法，其特征在于，在所述若判断两两为一组的小区块的数值同时大于或等于预设值，则将所述两两为一组的小区块作为背景区域并标记为关注标签之后，该方法包括：

若判断两两为一组的小区块的数值同时小于预设值，则将所述两两为一组的小区块作为非背景区域并标记为非关注标签。