[发明专利]一种图像处理方法及装置

权利要求书

1.一种图像处理方法，其特征在于，包括：

获取待处理图像，其中，所述待处理图像包括抠图目标；

通过金字塔特征提取网络对所述待处理图像进行处理，得到M个特征图像，其中，所述金字塔特征提取网络包括M个特征提取层，每个特征层的输入作为上一个特征层的输出，M为大于等于1的整数；

通过第一目标分割网络中的三元图图像分割网络对所述M个特征图像进行处理，并通过特征融合得到第一图像，其中，所述第一图像包括第一前景区域、第一背景区域及待确认区域，所述第一前景区域对应于所述抠图目标所对应的区域，所述待确认区域为所述第一前景区域与所述第一背景区域的交叠区域；

通过第二目标分割网络对所述M个特征图像进行处理，得到第二图像，其中，所述第二目标分割网络包括M+1个处理层，所述M+1个处理层中第一个处理层的输入为一个特征图像，所述第一个处理层后面的M-1个处理层的输入为上一层的输出和一个特征图像，最后的第M+1个处理层的输入为第M个处理层的输出；所述第二图像包括第二前景区域及第二背景区域，所述第二前景区域对应于所述抠图目标所对应的区域；

根据所述第一图像及所述第二图像，生成目标抠图图像，其中，所述目标抠图图像包括所述抠图目标。

2.如权利要求1所述的图像处理方法，其特征在于，所述第一目标分割网络包括M个残差通道注意力层及特征融合层；

所述通过第一目标分割网络中的三元图图像分割网络对所述M个特征图像进行处理，并通过特征融合得到第一图像，包括：

通过所述M个残差通道注意力层对所述M个特征图像进行处理，输出M个注意力矩阵；

通过所述特征融合层对所述M个注意力矩阵中的K个注意力矩阵进行处理，得到第一矩阵，其中，K为小于M的整数；

根据所述第一矩阵生成所述第一图像。

3.如权利要求2所述的图像处理方法，其特征在于，每个所述特征图像对应于一个特征矩阵；

所述通过所述M个残差通道注意力层对所述M个特征图像进行处理，输出M个注意力矩阵，包括：

将第i个特征矩阵及第i-1个特征矩阵作为第i个所述残差通道注意力层的输入，通过所述第i个残差通道注意力层输出第i个注意力矩阵，其中，所述第i个特征矩阵对应于第i个特征图像，所述第i-1个特征矩阵对应于第i-1个特征图像，i为大于1且小于M的整数；

将第j个特征矩阵作为第j个残差通道注意力层的输入，通过所述第j个残差通道注意力层输出第j个注意力矩阵，其中，所述第j个特征矩阵对应于第j个特征图像，j为1或M。

4.如权利要求2所述的图像处理方法，其特征在于，所述第一目标分割网络还包括计数器，所述计数器用于统计所述残差通道注意力层对特征图像的处理次数T，T为大于等于1的整数；所述M个特征图像对应M个特征矩阵；

所述通过所述M个残差通道注意力层对所述M个特征图像进行处理，输出M个注意力矩阵，包括：

获取所述计数器统计的所述处理次数T；

当所述处理次数小于第一阈值时，

将第一特征矩阵作为第一残差通道注意力层的输入，通过所述第一残差通道注意力层输出第一注意力矩阵，其中，所述第一特征矩阵对应于第一特征图像，所述第一特征图像根据所述金字塔特征提取网络中的第一特征提取层得到；

将第二特征矩阵及所述第一注意力矩阵作为第二残差通道注意力层的输入，通过所述第二残差通道注意力层输出第二注意力矩阵，其中，所述第二特征矩阵对应于第二特征图像，所述第二特征图像根据所述金字塔特征提取网络中的第二特征提取层得到；

将第三特征矩阵及所述第二注意力矩阵作为第三残差通道注意力层的输入，通过所述第三残差通道注意力层输出第三注意力矩阵，其中，所述第三特征矩阵对应于第三特征图像，所述第三特征图像根据所述金字塔特征提取网络中的第三特征提取层得到；

将第四特征矩阵作为第四残差通道注意力层的输入，通过所述第四残差通道注意力层输出第四注意力矩阵，其中，所述第四特征矩阵对应于第四特征图像，所述第四特征图像根据所述金字塔特征提取网络中的第四特征提取层得到。