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

权利要求书

1.一种图片的文案生成方法，其特征在于，所述方法包括：

获取待生成文案的图片；

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

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

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

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

2.如权利要求1所述的图片的文案生成方法，其特征在于，在所述将所述待生成文案的图片导入深度图片分类模型中之前，所述方法还包括：

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

3.如权利要求2所述的图片的文案生成方法，其特征在于，所述待生成文案的图片中记载的物体的装饰风格至少是美式、中式、欧式、日式、工业、轻奢、混搭、地中海、现代、古典、田园、东南亚中的一种风格。

4.如权利要求1所述的图片的文案生成方法，其特征在于，在所述将所述待生成文案的图片导入目标检测模型中之前，所述方法还包括：

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

5.如权利要求4所述的图片的文案生成方法，其特征在于，所述待生成文案的图片中记载物体的家居类型至少是床、柜、沙发、瓷砖、椅子中的一种。

6.如权利要求1所述的图片的文案生成方法，其特征在于，在所述将所述待生成文案的图片导入深度空间分类模型中之前，所述方法还包括：

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

7.如权利要求6所述的图片的文案生成方法，其特征在于，所述待生成文案的图片中记载物体的放置空间类型至少是阳台、卫生间、卧室、儿童房、玄关、厨房、客厅、餐厅、书房、衣帽间中的一种。

8.一种图片的文案生成装置，其特征在于，所述装置包括：

第一获取模块，用于获取待生成文案的图片；

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

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

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

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

9.一种图片的文案生成设备，其特征在于，所述设备包括：

处理器；以及

存储器，配置用于存储机器可读指令，所述指令在由所述处理器执行时，使得所述处理器执行如权利要求1-7任一项所述的图片的文案生成方法。

10.一种存储介质，其特征在于，所述存储介质存储有计算机程序，所述计算机程序被处理器执行如权利要求1-7任一项所述的图片的文案生成方法。