[发明专利]用于处理图像的方法和装置

说明书

本申请实施例公开了用于处理图像的方法和装置。该方法的一具体实施方式包括：获取包含至少两个对象的待处理图像；对待处理图像进行处理，以使待处理图像达到预设分辨率；将处理后的待处理图像输入至预先训练的图像语义分类模型中，得到分类图像，其中，图像语义分类模型用于表征输入的图像与分类图像之间的对应关系，分类图像为单通道颜色图像，包括的灰度值的数目等于所对应的输入的图像所包括的对象数；从所得到的分类图像中提取对象作为目标对象，并生成目标对象的图像。该实施方式提高了对目标对象从待处理图像中提取的准确率。

技术领域

本申请实施例涉及计算机技术领域，具体涉及互联网技术领域，尤其涉及用于处理图像的方法和装置。

背景技术

将图像中一些需要的部分从图像中提取出来，也称为抠图，为现有的图像处理中常用的处理方法之一。

现有的抠图技术一般采用图像分割的方法将特定对象从图像中提取出来，或者采用三元图的方法进行特定对象的提取。

发明内容

本申请实施例提出了用于处理图像的方法和装置。

第一方面，本申请实施例提供了一种用于处理图像的方法，该方法包括：获取包含至少两个对象的待处理图像；对待处理图像进行处理，以使待处理图像达到预设分辨率；将处理后的待处理图像输入至预先训练的图像语义分类模型中，得到分类图像，其中，图像语义分类模型用于表征输入的图像与分类图像之间的对应关系，分类图像为单通道颜色图像，包括的灰度值的数目等于所对应的输入的图像所包括的对象数；从所得到的分类图像中提取对象作为目标对象，并生成目标对象的图像。

在一些实施例中，对待处理图像进行处理，以使待处理图像达到预设分辨率，包括：响应于待处理图像沿第一方向的像素数目大于沿第二方向的像素数目，调整待处理图像沿第一方向的像素数目，以使待处理图像沿第一方向的像素数目达到预设像素数目，并生成第一图像，其中，第一方向与第二方向相交；确定预设像素数目与待处理图像沿第一方向的像素数目的第一比值；基于第一比值调整第一图像沿第二方向的像素数目，得到第二图像；基于第二图像，生成处理后的图像。

在一些实施例中，基于第二图像，生成处理后的图像，包括：响应于第二图像沿第二方向的像素数目小于预设像素数目，生成两张相同的增补图像，其中，各增补图像沿第一方向的像素数目与待处理图像沿第一方向的像素数目相同，各增补图像沿第二方向的像素数目为待处理图像沿第一方向的像素数目减去沿第二方向的像素数目的差值的一半，各增补图像的灰度值为预设样本图像集合中的样本图像的平均灰度值；沿第二方向，将两张相同的增补图像分别设置于第二图像的两侧，并与第二图像拼接；将拼接后的第二图像确定为处理后的图像。

在一些实施例中，图像语义分类模型为全卷积网络，全卷积网络包括卷积层以及反卷积层；以及将处理后的待处理图像输入至预先训练的图像语义分类模型中，得到分类图像，包括：将处理后的图像输入至卷积层，得到特征图像，其中，卷积层用于表征输入图像与特征图像之间的对应关系；将所得到的特征图像输入至反卷积层，得到分类图像，所得到的分类图像与处理后的待处理图像的尺寸相同。

在一些实施例中，图像语义分类模型通过以下方法训练得到：获取初始图像语义分类模型；获取包含有至少两个对象的样本图像的样本图像集合以及与每一个样本图像对应的分类样本图像，分类样本图像中包含有至少两个灰度值，每一个灰度值对应的颜色用于表征同一类别的对象，其中灰度值对应的颜色与对象类别之间的对应关系为预先标注的；将样本图像集合中的每一个样本图像作为输入，将与每一个样本图像对应的分类样本图像作为输出，训练初始图像语义分类模型，得到图像语义分类模型。

在一些实施例中，将样本图像集合中的每一个样本图像作为输入，将与每一个样本图像对应的分类样本图像作为输出，训练初始图像语义分类模型，得到图像语义分类模型，包括：利用反向传播算法以及随机梯度下降算法，对初始图像语义分类模型进行调整，使得初始图像语义分类模型所输出的图像的损失误差达到第一预设阈值；将所得到的初始图像语义分类模型确定为图像语义分类模型。