一种图像描述模型的生成方法、生成装置和存储介质

摘要

本申请公开了一种图像描述模型的生成方法、生成装置和存储介质，具体为获取图像样本和与图像样本对应的描述语句；将图像样本和描述语句输入第一预测模型，生成第一分布概率集合；获取图像样本中的各个实体对象对应的目标词汇对应的第二分布概率集合；分别为第一分布概率集合和第二分布概率集合分配权重值，以获取包含目标词汇在每个解码时刻对应的第三分布概率的第三分布概率集合，并输出预测语句；基于第一分布概率集合、第二分布概率集合和权重值，分别确定语句实体对象覆盖损失函数和语言序列损失函数，并确定模型损失函数，以对图像描述模型进行优化，直至生成图像描述模型。本申请实施例通过优化图像描述模型，提升图像描述的准确性。