摘要
本申请提供一种目标检测模型训练方法、装置及电子设备。本申请在获取样本图像以及所述样本图像对应的多张视觉提示图像的情况下,针对每一类别,通过特征提取网络对该类别对应的多个视觉提示图像进行特征提取,得到该类别对应的多个视觉提示特征,并通过特征聚合网络对该类别对应的参考特征和该类别对应的多个视觉提示特征进行特征聚合,得到该类别对应的聚合视觉提示特征,根据每一类别对应的聚合视觉提示特征对待训练模型进行训练,得到目标检测模型,通过在特征聚合网络中配置每一类别对应的参考特征,以使得针对每一类别,能够将该类别对应的多个提示特征聚合至该参考特征中,得到用于代表该类别的聚合视觉提示特征,提高了模型训练的效率。