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

说明书

本申请提供了一种图像处理方法以及相关装置。本申请实施例可应用于人工智能、智慧交通、辅助驾驶等各种场景。其方法包括：首先，获取包括抠图目标待处理图像；接着，通过金字塔特征提取网络对待处理图像进行处理，得到M个特征图像；然后，通过第一目标分割网络对M个特征图像进行处理，得到第一图像；并且，通过第二目标分割网络对M个特征图像进行处理，得到第二图像；最后，根据第一图像及第二图像，生成包括抠图目标的目标抠图图像。本申请提供的方法通过两个不同的目标分割网络对待处理图像的M个特征图像进行目标分割，得到第一图像及第二图像，通过第一图像及第二图像生成目标抠图图像，提高了抠图结果的准确性及抠图结果的精细程度。

技术领域

本申请涉及人工智能技术领域，尤其涉及一种图像处理方法及装置。

背景技术

随着计算机视觉的发展，越来越多的场合需要对图像进行分析处理，以实现从图像中获取需要的信息。例如，在人物照片处理时，需要通过抠图的方式，将人物从拍摄得到的人物照片中提取出来，通过抠取出的人物合成新的照片或者更换拍摄照片中人物的背景。

目前，图像抠图（Image matting）分为两类，一类是基于三元图（trimap）的抠图方法，另一类是无三元图（trimap-free）的抠图方法。基于三元图抠图方法所采用的算法需要人为标注来实现精细化抠图，无法实现全自动化的精细化抠图。无三元图的抠图方法包括两个阶段，第一阶段用分割网络产生一个粗糙分割图，第二阶段对分割图进行细化得到抠图结果；然而，在第一阶段产生的错误分割会影响第二阶段的图像细化，降低抠图结果的准确性及抠图结果的精细程度。

发明内容

本申请实施例提供了一种图像处理方法以及相关装置，通过两个不同的目标分割网络对待处理图像的M个特征图像进行目标分割，得到第一图像及第二图像，通过第一图像及第二图像生成目标抠图图像，提高了抠图结果的准确性及抠图结果的精细程度。

本申请的一方面提供一种图像处理方法，包括：

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

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

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

通过第二目标分割网络对M个特征图像进行处理，得到第二图像，其中，第二图像包括第二前景区域及第二背景区域，第二前景区域对应于抠图目标所对应的区域；

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

本申请的另一方面提供了一种图像处理装置，包括：

图像获取模块，用于获取待处理图像，其中，待处理图像包括抠图目标；

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

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

第二解码模块，用于通过第二目标分割网络对M个特征图像进行处理，得到第二图像，其中，第二图像包括第二前景区域及第二背景区域，第二前景区域对应于抠图目标所对应的区域；

抠图图像生成模块，用于根据第一图像及第二图像，生成目标抠图图像，其中，目标抠图图像包括抠图目标。