[发明专利]一种基于深度学习的影视海报自动生成方法与系统

权利要求书

1.一种基于深度学习的影视海报自动生成方法，其特征在于：

步骤一、获取用户提供的影视片单，以及用户为每一部影片提供的用于制作海报的非结构化数据；并将非结构化数据翻译成相应的海报制作模板；

步骤二、将所述的影视片单与自有的影视信息数据库进行影片的校验和确认；并根据所述的影视片单，利用互联网以及本地对应媒资视频进行素材图片的收集；

步骤三、将所述的收集到的素材图片输入到人脸检测模型，得到素材图片中的人脸数量和人脸坐标；根据得到的人脸坐标，截取对应的人脸图像，经过矫正后将其分别输入到人脸特征提取模型、人脸性别识别模型，分别得到人脸的特征向量以及每一张人脸对应的性别信息；

步骤四、将检测到的人脸的特征向量与自有的演员人脸特征向量库进行比对，得到图像中演员的姓名；并将素材图片输入到文本检测模型，得到文本的数量以及文本坐标；

步骤五、根据所述检测到的各素材图片中的人物数量、演员姓名、人脸坐标、文本数量以及文本坐标来筛除不符合步骤一得到的海报制作模板的要求的素材图片，得到精选素材图片；

步骤六、根据所述的海报制作模板、精选素材图片以及图像中的人脸坐标、文本坐标生成对应的影视海报；

步骤七、将所述的生成的多张影视海报输入到海报质量评估模型中，输出所述生成的影片海报的评估得分。

2.根据权利要求1所述的一种基于深度学习的影视海报自动生成方法，其特征在于：在步骤七执行后，将评估得分高于预设分值的影视海报输出，供使用者筛选。

3.根据权利要求1所述的一种基于深度学习的影视海报自动生成方法，其特征在于：步骤二中所述的影视信息数据库的获取方法，具体步骤如下：

（1）根据影片具有的基本属性以及特点建立影视信息数据库的表结构，表结构包括影片名称、影片上映年份、导演和演员；

（2）以所述影视信息数据库的表结构为指导，利用互联网获取影片公开的信息；

（3）对采集到的影片信息进行人工清洗、校对以及补充，得到整理过的影片信息；

（4）将整理过的影片信息入库，并定期执行步骤（1）至（3）来进行影片信息的更新。

4.根据权利要求1所述的一种基于深度学习的影视海报自动生成方法，其特征在于：步骤四中所述的演员人脸特征向量库的构建过程如下：

（1）以所述影视信息数据库中的演员字段为指导，利用互联网获取演员的公开文字信息和人脸头像图片；公开文字信息包含演员的姓名；

（2）将步骤（1）得到的演员人脸头像图片输入到人脸特征提取模型中，得到512维的演员的人脸特征向量；

（3）将所述512维的演员的人脸特征向量以及演员的姓名存储在关系型数据库中，得到所述演员人脸特征向量库。

5.根据权利要求1所述的一种基于深度学习的影视海报自动生成方法，其特征在于：步骤三中所述的人脸检测模型采用RetinaFace人脸检测算法；并选择ResNet50网络作为RetinaFace人脸检测算法的主干网络；其训练过程如下：

以开源的WIDERFACE人脸数据集以及自行标注的用于人脸检测的图像作为训练集和测试集；在人脸检测模型上使用所述的训练集进行训练，并保存训练过程中生成的模型权重值文件以及模型网络结构文件，得到可用的人脸检测模型。

6.根据权利要求1所述的一种基于深度学习的影视海报自动生成方法，其特征在于：步骤三中所述的人脸特征提取模型采用ResNet100网络为模型的主干构建的ResNet100-ArcFace人脸特征提取模型，其训练过程如下：

以开源的LFW以及MegaFace人脸数据集作为训练集和测试集；在人脸特征提取模型上使用训练集进行训练，并保存训练过程中生成的模型权重值文件以及模型网络结构文件，得到可用的人脸特征提取模型。