跨媒体检索模型的训练方法、跨媒体检索方法及其设备
摘要:
本发明提供了一种跨媒体检索模型的训练方法、跨媒体检索方法及其设备,其中,该训练方法包括获取跨媒体数据集,每条数据包括图像、文本、时间、作者及语义标签;获取每条数据的语义标签向量及各部分的特征向量;计算标签相似度,得到相似度矩阵;由图像、时间及作者特征向量得到第一模态数据,由文本、时间及作者特征向量得到第二模态数据;将两模态数据分别输入至特征映射网络,得到第一特征映射向量和第二特征映射向量;计算两特征映射向量的相似度,得到相似度矩阵;计算两个相似度矩阵的差异值,将两个特征映射向量输入至模态判别网络,得到对抗损失函数的值,优化网络参数,得到跨媒体检索模型。通过上述方案能准确进行语义关联表达。
0/0