[发明专利]图像生成方法、装置、电子设备及存储介质

说明书

本申请实施例公开了一种图像生成方法、装置、电子设备及存储介质，该图像生成方法通过先对参考场景图像进行边缘检测得到第一边缘图像，通过图像变换处理快捷地得到多个第二边缘图像，相应地可以基于场景图像生成模型调整第二边缘图像的图像类型得到多个目标场景图像，进而可生成多个目标样本图像。因此，只需采集少量的参考场景图像和基准图像即可生成大量的目标样本图像，相较于采集大量的基准图像的方式，在生成相同数量的目标样本图像的前提下，本申请实施例提供的图像生成方法能够有效降低图像采集数量，降低目标样本图像的生成成本，提高目标样本图像的生成效率，可以广泛应用于人工智能、图像处理、云计算等技术领域。

技术领域

本申请涉及人工智能技术领域，特别是涉及一种图像生成方法、装置、电子设备及存储介质。

背景技术

随着人工智能技术的快速发展，各类图像处理模型的应用也变得越来越广泛。在模型应用前需要对该图像处理模型进行训练，以使得图像处理模型的性能达到最佳。为了提高模型的训练效果，一般需要大量的训练图像来对图像处理模型进行训练。除了上述场景以外，其他场景也可能有大量的图像需求，例如在进行图像处理软件测试时也需要使用到大量的测试图像。

目前，目标样本图像(例如训练图像或者测试图像等)可以基于采集真实的基准图像生成，相比于人工收集，可以在一定程度上降低图像的采集成本。然而，相关技术中，基准图像与目标样本图像之间一般是一一对应的关系，生成目标样本图像时仍然需要采集大量的基准图像，目标样本图像的生成效率仍有待提高。

发明内容

以下是对本文详细描述的主题的概述。本概述并非是为了限制权利要求的保护范围。

本申请实施例提供了一种图像生成方法、装置、电子设备及存储介质，能够提高目标样本图像的生成效率。

一方面，本申请实施例提供了一种图像生成方法，包括：

获取参考场景图像，对所述参考场景图像进行边缘检测，得到所述参考场景图像对应的第一边缘图像；

对所述第一边缘图像进行图像变换处理，得到所述第一边缘图像对应的多个第二边缘图像；

将所述第二边缘图像输入至预设的场景图像生成模型，基于所述场景图像生成模型调整所述第二边缘图像的图像类型，得到所述第二边缘图像对应的目标场景图像；

获取基准图像，将所述基准图像与各个所述目标场景图像进行合成处理，生成多个目标样本图像。

另一方面，本申请实施例还提供了一种图像生成装置，包括：

边缘检测模块，用于获取参考场景图像，对所述参考场景图像进行边缘检测，得到所述参考场景图像对应的第一边缘图像；

变换模块，用于对所述第一边缘图像进行图像变换处理，得到所述第一边缘图像对应的多个第二边缘图像；

模型处理模块，用于将所述第二边缘图像输入至预设的场景图像生成模型，基于所述场景图像生成模型调整所述第二边缘图像的图像类型，得到所述第二边缘图像对应的目标场景图像；

合成模块，用于获取基准图像，将所述基准图像与各个所述目标场景图像进行合成处理，生成多个目标样本图像。

进一步，所述场景图像生成模型包括多个依次连接的卷积层和多个依次连接的反卷积层，上述模型处理模块具体用于：

基于多个所述卷积层对所述第二边缘图像进行卷积处理，得到目标卷积图像；

基于多个所述反卷积层对所述目标卷积图像进行反卷积处理，得到所述第二边缘图像对应的目标场景图像。

进一步，所述卷积层的数量和所述反卷积层的数量相同，上述模型处理模块具体用于：