[发明专利]显著性图像生成方法、装置及服务器

说明书

本发明提供一种显著性图像生成方法、装置及服务器，所述方法包括：获得所有测试用户的遗忘度特征，并根据测试用户的遗忘度特征将测试用户分为多个群体，其中每个群体的遗忘度特征相同；获取清晰样本图像集以及每个群体的标注样本图像集，其中标注样本图像集与清晰样本图像集对应；采用神经网络模型分别每个群体的标注样本图像集以及清晰样本图像进行训练，构建每个群体的预测模型；获得目标用户的遗忘度特征，根据目标用户的遗忘度特征以及预测模型确定目标预测模型，并将目标图像输入目标预测模型中，以生成显著性图像，提高了生成的显著性图像与用户实际的视觉系统的显著性的匹配度。

技术领域

本发明涉及计算机视觉领域，尤其涉及一种显著性图像生成方法、装置及服务器。

背景技术

人类视觉系统的显著性具有快速搜索和定位感兴趣目标的能力，人在面对自然场景时，可根据视觉系统的显著性快速地过滤掉不重要的信息，让我们的注意力更加集中在感兴趣的区域。因此，在图像推送的应用场景中，通过预测不同用户的视觉系统的显著性，并根据预测的图像显著性预测结果生成待推送的图像，提高了用户对于待推送的图像的感兴趣程度。

现有技术中，显著性检测方法大多采用深度学习的方法，根据用户提交的显著性图像样本进行训练，即将卷积操作与池化操作相结合，利用深度卷积神经网络提取图像特征，获得显著性预测模型，并根据显著性预测模型获得待推送的显著性图像。

然而，现有技术中，用户对于图像的遗忘性会导致用户提交的显著性图像样本中丢失一些图像的特征信息，使得获得显著性预测模型的准确度较低，导致根据生成的显著性图像与用户实际的视觉系统的显著性不匹配。

发明内容

本发明的目的在于提供一种显著性图像生成方法、装置及服务器，以提高生成的显著性图像与用户实际的视觉系统的显著性的匹配度。

第一方面，本发明提供一种显著性图像生成方法，包括：

获得所有测试用户的遗忘度特征，并根据所述测试用户的遗忘度特征将所述测试用户分为多个群体，其中每个群体的遗忘度特征相同；

获取清晰样本图像集以及每个群体的标注样本图像集，其中所述标注样本图像集与所述清晰样本图像集对应；

采用神经网络模型分别所述每个群体的标注样本图像集以及所述清晰样本图像进行训练，构建每个群体的预测模型；

获得目标用户的遗忘度特征，根据所述目标用户的遗忘度特征以及所述预测模型确定目标预测模型，并将目标图像输入所述目标预测模型中，以生成显著性图像。

在一种可能的设计中，所述根据所述测试用户的遗忘度特征将所述测试用户分为多个群体，包括：

根据所述测试用户的遗忘度特征将所述测试用户分为第一群体、第二群体以及第三群体，其中所述第一群体中的测试用户的遗忘度较高、所述第二群体中的测试用户的遗忘度正常以及所述第三群体中的测试用户的遗忘度较低。

在一种可能的设计中，所述采用神经网络模型分别所述每个群体的标注样本图像集以及所述清晰样本图像进行训练，构建每个群体的预测模型，包括：

分别根据所述第一群体、所述第二群体以及所述第三群体的标注样本图像集以及清晰样本图像集获得第一训练集、第二训练集以及第三训练集，并采用神经网络模型分别对所述第一训练集、所述第二训练集以及所述第三训练集进行卷积神经网络训练，构建第一预测模型、第二预测模型以及第三预测模型；

其中所述卷积神经网络训练的步骤为：

构建卷积神经网络模型，并将所述清晰样本图像集输入所述卷积神经网络模型中进行迭代训练，获得显著性预测图集；

根据所述标注样本图像集与所述显著性预测图集确定损失函数集，并根据损失函数集中的最小损失函数确定预测模型。