图像边框识别模型训练方法及装置、电子设备和介质

本公开提供了一种图像边框识别模型训练方法、装置、电子设备、计算机可读存储介质和计算机程序产品，涉及人工智能领域，尤其涉及深度学习、图像识别、信息推荐技术领域。实现方案为：获取待识别的图像，用于在多个分类子任务下进行边框识别；将待识别的图像输入主干网络以获得第一图像特征；对于每个分类子任务，执行以下操作：将第一图像特征输入该分类子任务所对应的全连接层获得第一特征向量；将该分类子任务对应的第一特征向量作为关键向量和数值向量、至少一个其他分类子任务对应的第一特征向量作为查询向量输入该分类子任务对应的多头注意力模块，获得在该分类子任务下待识别的图像的边框信息属于各个类别的概率，以基于概率确定边框信息。