图像描述语句生成模块的训练方法及装置、电子设备

摘要

一种图像描述语句生成模块的训练方法及装置、电子设备，该方法包括：获取第一样本图像及其第一描述语句，并将第一样本图像输入预先训练的预测模块中进行标签识别，得到至少一个第一特征标签及其第一概率向量。获取第一特征标签的标签向量，并根据第一特征标签的标签向量和第一概率向量，生成第一特征标签的特征信息。最后，根据特征信息和第一描述语句对生成式模型进行训练，得到语句生成模块，既引入对特征标签的重要性分析，又基于特征信息的可导性实现误差量的梯度反传，达到更优训练效果，提升图像描述的准确性。