[发明专利]一种图像的风格模式推荐方法和终端

说明书

本申请公开了一种图像的风格模式推荐方法和终端，用以减少用户选择风格模式的时间，从而提升用户使用体验。本申请方法包括：将待处理图像输入到已训练的场景识别神经网络模型中，获取所述场景识别神经网络模型输出的至少一个场景以及所述场景对应的置信度；根据预设的场景与风格模式的对应关系，确定获取的所述场景对应的风格模式；根据所述置信度对确定的风格模式进行排序，并通过显示屏进行显示；确定用户选择的风格模式，并使用所述用户选择的风格模式对应的图片参数对所述待处理图像进行风格化处理。这样，根据待处理图像的场景，为用户推荐适合待处理图像的风格模式，可以减少用户选择风格模式的时间，提升用户使用体验。

技术领域

本申请涉及图像处理领域，尤其涉及一种图像的风格模式推荐方法和终端。

背景技术

图像风格迁移技术是对图像进行风格化处理，使其获得风格图片的艺术风格，从而将普通的拍照图片变成了艺术作品。但现有技术的做法是将大量的风格模式按固定顺序排列在用户界面上，让用户自行选择一种风格，然后软件根据用户的选择进行相应的风格化处理。用户需要在多种风格模式中不断选择才能得到较为满意的结果。

因此，现有的风格模式推荐方法会导致用户耗费大量时间来选择合适的风格模式。

发明内容

本申请实施例提供一种图像的风格模式推荐方法和终端，用以减少用户选择合适的风格模式的时间，提升用户使用体验。

第一方面，本申请实施例提供一种图像的风格模式推荐方法，该方法包括：

将待处理图像输入到已训练的场景识别神经网络模型中，获取所述场景识别神经网络模型输出的至少一个场景以及所述场景对应的置信度；其中，所述待处理图像为通过摄像头获取的图像帧或本地存储的图像；

根据预设的场景与风格模式的对应关系，确定获取的所述场景对应的风格模式；

根据所述置信度对确定的风格模式进行排序，并通过显示屏进行显示；

确定用户选择的风格模式，并使用所述用户选择的风格模式对应的图片参数对所述待处理图像进行风格化处理。

上述方法，通过对待处理图像进行场景识别，确定待处理图像包含的场景，并根据确定的场景对风格模式进行排序显示。这样，根据待处理图像的场景，为用户推荐适合待处理图像的风格模式，可以减少用户选择风格模式的时间，提升用户使用体验。

在一种可能的实现方式中，若一个场景对应至少两种风格模式，则确定获取的所述场景对应的风格模式之后，通过显示屏进行显示之前，所述方法还包括：

获取所述待处理图像的内容特征以及确定的所述风格模式的内容特征；

针对同一场景对应的任意一种风格模式，将所述风格模式的内容特征与所述待处理图像的内容特征进行对比，得到所述风格模式与所述待处理图像的相似度；

根据同一场景对应的每种所述风格模式与所述待处理图像的相似度，对所述同一场景对应的每种风格模式进行排序。

上述方法，限定了一个场景对应多种风格模式的情况下时，通过对比风格模式的内容特征，为同一场景下的风格模式进行排序。这样，可以使推荐的风格模式更加准确，从而进一步提高用户的操作体验。

在一种可能的实现方式中，获取所述待处理图像的内容特征，包括：

将所述待处理图像输入到已训练的特征提取神经网络模型中，获取所述特征提取神经网络模型输出的所述待处理图像的内容特征；或；

将所述场景识别神经网络模型中特征提取层的输出结果作为所述待处理图像的内容特征。