[发明专利]图像处理方法、装置、计算机设备和存储介质

说明书

本申请涉及一种图像处理方法、装置、计算机设备和存储介质。所述方法包括：获取原始图像和目标生成尺寸；按照目标生成尺寸的高度和宽度的比值对原始图像进行拓展，得到拓展图像；根据拓展图像的拓展区域和原始区域生成掩码图像，掩码图像与拓展图像维度相同；将掩码图像、目标生成尺寸和原始图像输入图像生成网络，输出生成图像，生成图像的尺寸与目标生成尺寸相同，图像生成网络为目标对抗网络的子网络。通过训练好的图像生成网络对原始图像特征提取和重构，得到与展示设备相适应的图像，使得生成的图像具备更完整的图像信息，播放该图像，可以提升用户体验。

技术领域

本申请涉及计算机技术领域，尤其涉及一种图像处理方法、装置、计算机设备和存储介质。

背景技术

视频作为娱乐媒体的重要载体，其承载着重要信息。随着自媒体以及UGC的兴起，越来越多的用户使用自己的拍摄设备记录生活并上传至互联网。

拍摄设备拍摄视频时，其中大量的视频时是竖屏拍摄的，用户将视频上传到互联网，播放视频时，经常是横屏播放。目前将视频从竖转横，通常采用显著性检测方法检测视频主体，对视频进行裁剪出主体目标，裁剪视频主体部分，造成信息丢失，影响观感。

发明内容

为了解决上述技术问题，本申请提供了一种图像处理方法、装置、计算机设备和存储介质。

第一方面，本申请提供了一种图像处理方法，包括：

获取原始图像和目标生成尺寸；

按照目标生成尺寸的高度和宽度的比值对原始图像进行拓展，得到拓展图像，拓展图像包括拓展区域和原始图像对应的原始区域；

根据拓展图像的拓展区域和原始区域生成掩码图像，掩码图像与拓展图像维度相同；

将掩码图像、目标生成尺寸和原始图像输入图像生成网络，输出生成图像，生成图像的尺寸与目标生成尺寸相同，图像生成网络为目标对抗网络的子网络。

第二方面，本申请提供了一种图像处理装置，包括：

数据获取模块，用于获取原始图像和目标生成尺寸；

图像拓展模块，用于按照目标生成尺寸的高度和宽度的比值对原始图像进行拓展，得到拓展图像，拓展图像包括拓展区域和原始图像对应的原始区域；

掩码图像生成模块，用于根据拓展图像的拓展区域和原始区域生成掩码图像，掩码图像与拓展图像维度相同；

图像生成模块，用于将掩码图像、目标生成尺寸和原始图像输入图像生成网络，输出生成图像，生成图像的尺寸与目标生成尺寸相同，图像生成网络为目标对抗网络的子网络。

一种计算机设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现以下步骤：

获取原始图像和目标生成尺寸；

按照目标生成尺寸的高度和宽度的比值对原始图像进行拓展，得到拓展图像，拓展图像包括拓展区域和原始图像对应的原始区域；

根据拓展图像的拓展区域和原始区域生成掩码图像，掩码图像与拓展图像维度相同；

将掩码图像、目标生成尺寸和原始图像输入图像生成网络，输出生成图像，生成图像的尺寸与目标生成尺寸相同，图像生成网络为目标对抗网络的子网络。

一种计算机可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时实现以下步骤：

获取原始图像和目标生成尺寸；

按照目标生成尺寸的高度和宽度的比值对原始图像进行拓展，得到拓展图像，拓展图像包括拓展区域和原始图像对应的原始区域；