[发明专利]图像颜色编辑方法及装置

摘要

本发明提供了一种图像颜色编辑方法及装置，所述方法通过用户输入的不同类别的用户交互信息，确定待编辑图像中每一类别的种子像素集对应的距离图，并将每一距离图作为待编辑图像的一个新的通道图，结合待编辑图像的RGB三个通道图，确定预设数量的多通道图像子块。基于训练好的多特征联合网络模型，提取每一多通道图像子块中的深度特征，根据深度特征，得到待编辑图像中每一像素所属类别的概率。最后根据得到的概率，并基于最小化的图像分割能量函数，确定待编辑图像中每一像素所属类别，并对所述待编辑图像的颜色进行编辑。本实施例提供的图像颜色编辑方法，有效的结合待编辑图像中像素的距离信息和颜色信息，从而保证了图像颜色编辑的准确性。

主权项

1.一种图像颜色编辑方法，其特征在于，包括：S1，基于不同类别的用户交互信息，提取待编辑图像上不同类别的像素以构成不同类别的种子像素集，并根据所述待编辑图像上每一像素与每一类别的种子像素集中每一种子像素之间的空间距离，确定每一类别的种子像素集对应的所述待编辑图像的距离图；S2，将每一个所述距离图作为所述待编辑图像的一个新的通道图，并结合所述待编辑图像的RGB三个通道图，确定所述待编辑图像的多通道图像，并从所述多通道图像中预设位置处提取预设数量的多通道图像子块，所述预设位置为所述多通道图像中与每一类别的用户交互信息对应的位置；S3，基于训练好的多特征联合网络模型，提取每一多通道图像子块中的深度特征，根据所述深度特征，得到所述待编辑图像中每一像素所属类别的概率；S4，根据得到的所述待编辑图像中每一像素所属类别的概率以及最小化的图像分割能量函数，确定所述待编辑图像中每一像素所属类别，并根据所述待编辑图像中每一像素所属类别，对所述待编辑图像的颜色进行编辑。