[发明专利]一种面向主题的图像场景图生成方法

摘要

本发明公开了一种面向主题的图像场景图生成方法。本发明首先使用LDA概率主题模型，提取文本的主题概率分布以及主题的词汇分布，构建图像主题数据集。然后利用图像主题数据集，预训练得到图像主题预测模型。其次，模型对图像进行目标检测，并融合全局特征、目标特征、类别特征以及主题特征得到关系特征，对检测的目标预测其属性以及目标间的关系，训练得到场景图生成模型。使用U‑2‑Net对图像进行显著性检测，并利用检测结果得到图像中各像素点的显著值，通过显著值得到图像的显著性区域。结合显著性区域以及生成的图像场景图，根据定义规则最终生成面向主题的图像场景图。本发明解决了现有场景图中存在主题无关目标的问题。