[发明专利]图片的文案生成方法、装置、设备及存储介质

说明书

本申请提供一种图片的文案生成方法、装置、设备及存储介质，其中，图片的文案生成方法包括步骤：将待生成文案的图片导入深度图片分类模型中，以获取深度图片分类模型输出的待生成文案的图片的风格信息，将待生成文案的图片导入深度空间分类模型中，以获取深度空间分类模型输出的待生成文案的图片的空间分类信息，将待生成文案的图片导入目标检测模型中，以获取目标检测模型输出的待生成文案的图片的类型信息，将空间分类信息、类型信息、风格信息导入关键词生成模型中，以获取关键词生成模型输出的待生成文案的图片对应的文案信息。通过本申请生成的图片对应的文案具有多样性。

技术领域

本申请涉及计算机领域，具体而言，涉及一种图片的文案生成方法、装置、设备及存储介质。

背景技术

文案生成领域应用前景广阔，具有巨大的市场需求，例如电商中的商品推荐语、新闻财报等，这可以大大提高效率以及降低一些重复性的工作。

然而，现有的从图片到文本的自动生成技术主要是基于端到端的Seq2Seq算法模型框架实现，这种方式生成的文案较为单一，而无法生成大量的多样性文本。

发明内容

本申请实施例的目的在于提供一种图片的文案生成方法、装置、设备及存储介质，用以实现基于图片生成的文案具有多样性。

为此，本申请第一方面提供一种图片的文案生成方法，所述方法包括步骤：

获取待生成文案的图片；

将所述待生成文案的图片导入深度图片分类模型中，以获取所述深度图片分类模型输出的所述待生成文案的图片的风格信息，所述待生成文案的图片的风格信息表征所述待生成文案的图片中记载的物体的装饰风格；

将所述待生成文案的图片导入深度空间分类模型中，以获取所述深度空间分类模型输出的所述待生成文案的图片的空间分类信息，所述待生成文案的图片的空间分类信息表征所述待生成文案的图片中记载物体的放置空间类型；

将所述待生成文案的图片导入目标检测模型中，以获取所述目标检测模型输出的所述待生成文案的图片的类型信息，所述待生成文案的图片的类型信息表征所述待生成文案的图片中记载物体的家居类型；

将所述空间分类信息、所述类型信息、所述风格信息导入关键词生成模型中，以获取所述关键词生成模型输出的所述待生成文案的图片对应的文案信息。

在本申请第一方面中，通过分别将待生成文案的图片导入训练好的目标检测模型、深度图片分类模型、深度空间分类模型中，可得到待生成文案的图片对应的空间分类信息、类型信息、风格信息，这样一来就能够得到待生成文案的图片的更多信息，进而能够基于待生成文案的图片的更多信息生成更加多样的文案，且可提高所生成文案的可读性、通顺性、相关性。

在本申请第一方面中，作为一种可选的实施方式，在所述将所述待生成文案的图片导入深度图片分类模型中之前，所述方法还包括：

根据第一预设训练样本训练ResNet神经网络，并得到所述深度图片分类模型。

在本可选的实施方式中，通过第一预设训练样本可训练ResNet神经网络，进而能够得到深度图片分类模型。

在本申请第一方面中，作为一种可选的实施方式，所述待生成文案的图片中记载的物体的装饰风格至少是美式、中式、欧式、日式、工业、轻奢、混搭、地中海、现代、古典、田园、东南亚中的一种风格。

本可选的实施方式中能够识别多种待生成文案的图片中记载的物体的装饰风格。

在本申请第一方面中，作为一种可选的实施方式，在所述将所述待生成文案的图片导入目标检测模型中之前，所述方法还包括：

根据第二预设训练样本训练YOLO神经网络模型，并得到所述目标检测模型。