[发明专利]图像处理方法及装置

权利要求书

1.一种图像处理方法，包括：

接收图像处理请求，并基于所述图像处理请求确定待识别图像；

基于图像主体识别规则识别所述待识别图像的主体图像；

根据所述待识别图像和所述主体图像获得待处理图像，并将所述待处理图像输入至图像布局模型，获得所述待识别图像对应的图像布局信息。

2.如权利要求1所述的方法，所述图像布局模型基于如下步骤训练获得：

获取创意图像，其中，所述创意图像包括实际布局信息；

基于所述创意图像获得创意原图像和创意主体图像，并基于所述创意原图像和所述创意主体图像获得待处理训练图像；

将所述待处理训练图像输入至图像布局模型，获得所述图像布局模型输出的预测布局信息；

基于所述预测布局信息和所述实际布局信息计算模型损失值；

根据所述模型损失值调整所述图像布局模型的模型参数，并继续训练，直至达到模型训练停止条件。

3.如权利要求2所述的方法，所述基于所述创意图像获得创意原图像和创意主体图像，包括：

基于所述创意图像获得创意原图像；

根据所述创意原图像获得创意主体图像。

4.如权利要求3所述的方法，基于所述创意图像获得创意原图像，包括：

获取创意图像，并确定所述创意图像中的实际布局信息，其中，所述实际布局信息是对所述创意图像进行标注获得；

根据所述实际布局信息对所述创意图像进行遮挡，获得遮挡创意图像；

将所述遮挡创意图像输入至图像修复模型，获得所述创意图像对应的创意原图像。

5.如权利要求3所述的方法，基于所述创意原图像获得创意主体图像，包括：

识别所述创意原图像中的主体区域和背景区域；

将所述主体区域中的像素点设置为第一颜色信息，将所述背景区域中的像素点设置为第二颜色信息，获得创意主体图像。

6.如权利要求2所述的方法，所述方法还包括：

将所述预测布局信息输入至布局判别模型，获得判别结果；

基于所述判别结果调整所述图像布局模型的模型参数，并继续训练，直至达到模型训练条件。

7.如权利要求1所述的方法，所述基于图像主体识别规则识别所述待识别图像的主体图像，包括：

识别所述待识别图像中的主体区域和背景区域；

将所述主体区域中的像素点设置为第一颜色信息，将所述背景区域中的像素点设置为第二颜色信息，获得所述待识别图像的主体图像。

8.如权利要求7所述的方法，所述根据待识别图像和所述主体图像获得待处理图像，包括：

获取所述待识别图像中每个像素点的三原色信息和所述主体图像中每个像素点的主体颜色信息，其中，所述主体颜色信息包括第一颜色信息或第二颜色信息；

基于每个像素点的三原色信息和主体颜色信息生成每个像素点的像素颜色信息；

根据每个像素点的像素颜色信息获得所述待处理图像。

9.如权利要求1所述的方法，所述图像布局模型中包含特征提取器、编码器和解码器；

所述将所述待处理图像输入至图像布局模型，获得所述待识别图像对应的图像布局信息，包括：

将所述待处理图像输入至所述特征提取器做特征提取处理，获得所述待处理图像对应的图像特征；

将所述图像特征输入至所述编码器做编码处理，获得图像编码矩阵；

将所述图像编码矩阵输入至所述解码器做解码处理，获得图像解码矩阵；

将所述图像解码矩阵做归一化处理，获得所述图像解码矩阵对应的图像布局信息。