[发明专利]摆盘图像的生成方法、图像生成模型的训练方法及装置

说明书

本申请实施例提供了一种摆盘图像的生成方法、图像生成模型的训练方法及装置，涉及机器学习和图像识别技术领域。所述摆盘图像的生成方法包括：获取目标餐品的原始图像；基于原始图像获取目标餐品的餐品信息，餐品信息用于指示目标餐品的相关特征；基于原始图像和餐品信息，生成目标餐品的摆盘图像。本申请实施例提供的技术方案，使得用户可以根据自动生成的摆盘图像，对目标餐品进行摆盘，相比于对各个餐品手动设计对应的摆盘方案，降低了确定摆盘方案所需的成本。

技术领域

本申请实施例涉及机器学习和图像识别技术领域，特别涉及一种摆盘图像的生成方法、图像生成模型的训练方法及装置。

背景技术

随着网络点餐的发展，摆盘越好看的餐品所获得的关注越多。

在相关技术中，需要相关技术人员按照自身经验，对各个餐品分别手动设计对应的摆盘方案，采用这种方式确定摆盘方案的成本较高。

发明内容

本申请实施例提供了一种摆盘图像的生成方法、图像生成模型的训练方法及装置，能够降低确定摆盘方案的成本。所述技术方案如下：

根据本申请实施例的一个方面，提供了一种摆盘图像的生成方法，所述方法包括：

获取目标餐品的原始图像；

基于所述原始图像获取所述目标餐品的餐品信息，所述餐品信息用于指示所述目标餐品的相关特征；

基于所述原始图像和所述餐品信息，生成所述目标餐品的摆盘图像。

根据本申请实施例的一个方面，提供了一种图像生成模型的训练方法，所述方法包括：

获取样本餐品图像以及所述样本餐品图像对应的餐品信息，所述餐品信息用于指示所述样本餐品图像中包含的样本餐品的相关特征；

获取所述样本餐品的参考摆盘图像；

通过图像生成模型基于所述样本餐品图像和所述餐品信息，生成预测摆盘图像；

基于所述样本餐品图像、所述参考摆盘图像、以及所述预测摆盘图像，确定所述图像生成模型的损失；

基于所述图像生成模型的损失，调整所述图像生成模型的参数。

根据本申请实施例的一个方面，提供了一种摆盘图像的生成装置，所述装置包括：

图像获取模块，用于获取目标餐品的原始图像；

信息获取模块，用于基于所述原始图像获取所述目标餐品的餐品信息，所述餐品信息用于指示所述目标餐品的相关特征；

图像生成模块，用于基于所述原始图像和所述餐品信息，生成所述目标餐品的摆盘图像。

根据本申请实施例的一个方面，提供了一种图像生成模型的训练装置，所述装置包括：

信息获取模块，用于获取样本餐品图像以及所述样本餐品图像对应的餐品信息，所述餐品信息用于指示所述样本餐品图像中包含的样本餐品的相关特征；

图像获取模块，用于获取所述样本餐品的参考摆盘图像；

图像生成模块，用于通过图像生成模型基于所述样本餐品图像和所述餐品信息，生成预测摆盘图像；

损失确定模块，用于基于所述样本餐品图像、所述参考摆盘图像、以及所述预测摆盘图像，确定所述图像生成模型的损失；

参数调整模块，用于基于所述图像生成模型的损失，调整所述图像生成模型的参数。