[发明专利]图像主色提取方法、装置、电子设备及存储介质

说明书

本发明提供一种图像主色提取方法、装置、电子设备及存储介质，涉及图像处理技术领域。该图像主色提取方法包括：将待提取图像的色彩空间，从第一色彩空间转换为第二色彩空间。根据第二色彩空间的色彩信息进行聚类，获取色彩聚类结果，其中，色彩聚类结果包括：多个色彩类，每个色彩类中颜色之间的相似度在预设范围内。根据色彩聚类结果，从待提取图像中确定色彩聚类结果对应的局部区域。在第一色彩空间中，对局部区域进行直方图取色，获取待提取图像中的图像主色。由于是在聚类的结果中进行直方图取色，考虑了颜色之间的相似性，排除了颜色纯度高但是面积小的区域的影响，提高了提取图像主色的准确度，增强了提取图像主色时的鲁棒性。

技术领域

本发明涉及图像处理技术领域，具体而言，涉及一种图像主色提取方法、装置、电子设备及存储介质。

背景技术

图像中，颜色是最常用且最具有价值的一种特征信息，在图像处理、图像检索中被广泛应用。因此，需要快速准确的提取目标图像中的颜色信息。

现有技术中，最常用的是直接采用颜色直方图提取算法，通过颜色直方图对图像中的颜色信息进行提取，获取图像的主色信息。

但是，由于颜色直方图是描述不同的色彩在整个图像中所占的比例，并未考虑到颜色之间的相似性，忽略了视觉感受，因此，获取的主色信息，往往会受到颜色纯度高而面积小的区域的影响。

发明内容

本发明的目的在于，针对上述现有技术中的不足，提供一种图像主色提取方法、装置、电子设备及存储介质，以解决获取的图像主色不准确的问题。

为实现上述目的，本发明实施例采用的技术方案如下：

第一方面，本发明实施例提供了一种图像主色提取方法，包括：将待提取图像的色彩空间，从第一色彩空间转换为第二色彩空间。根据第二色彩空间的色彩信息进行聚类，获取色彩聚类结果，其中，色彩聚类结果包括：多个色彩类，每个色彩类中颜色之间的相似度在预设范围内。根据色彩聚类结果，从待提取图像中确定色彩聚类结果对应的局部区域。在第一色彩空间中，对局部区域进行直方图取色，获取待提取图像中的图像主色。

可选地，对第二色彩空间的色彩信息进行聚类，获取色彩聚类结果，包括：在第二色彩空间中，平均获取N个样本点。将待提取图像中的每个像素点，归属到每个像素点的颜色对应的样本点所在的颜色区域内，得到N个样本区域。将每个样本区域的几何中心作为迭代样本点，进行迭代，以确定每个像素点的颜色对应的迭代样本点以及迭代样本区域。若迭代样本点或迭代次数满足预设条件，则停止迭代，并根据迭代样本点以及迭代样本区域，获取聚类结果。

可选地，若迭代样本点或迭代次数满足预设条件，则停止迭代，包括：若每个所述迭代样本点与对应的前一迭代样本点之间的距离，均在预设范围内，则停止迭代。或者，若迭代次数达到预设的迭代次数，则停止迭代。

可选地，将待提取图像中的每个像素点，归属到每个像素点对应的样本点，得到N个样本区域，包括：获取待提取图像中每个像素点与每个样本点之间的距离。将每个像素点归属到与像素点距离最近的样本点，得到样本区域。

可选地，在第一色彩空间中，对局部区域进行直方图取色，获取待提取图像中的图像主色，包括：将第一色彩空间按照预设步长进行划分，得到简化色彩空间，其中，简化色彩空间包括多个色彩区域，每个色彩区域的容量根据预设步长确定。在简化色彩空间中，对局部区域进行直方图取色，得到简化色彩空间的主色。在第一色彩空间中，对简化色彩空间的主色进行直方图取色，获取待提取图像的图像主色。

可选地，在简化色彩空间，对局部区域进行直方图取色，得到简化色彩空间的主色，包括：遍历待提取图像的每个像素点，根据每个像素点的颜色，将每个像素点归属到每个像素点对应的色彩区域。将多个色彩区域中像素点最多的一个色彩区域的颜色作为简化色彩空间的主色。