目标检测模型训练方法、装置及电子设备

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
目标检测模型训练方法、装置及电子设备
申请号:CN202510191809
申请日期:2025-02-20
公开号:CN119693775B
公开日期:2025-05-02
类型:发明专利
摘要
本申请提供一种目标检测模型训练方法、装置及电子设备。本申请在获取样本图像以及所述样本图像对应的多张视觉提示图像的情况下,针对每一类别,通过特征提取网络对该类别对应的多个视觉提示图像进行特征提取,得到该类别对应的多个视觉提示特征,并通过特征聚合网络对该类别对应的参考特征和该类别对应的多个视觉提示特征进行特征聚合,得到该类别对应的聚合视觉提示特征,根据每一类别对应的聚合视觉提示特征对待训练模型进行训练,得到目标检测模型,通过在特征聚合网络中配置每一类别对应的参考特征,以使得针对每一类别,能够将该类别对应的多个提示特征聚合至该参考特征中,得到用于代表该类别的聚合视觉提示特征,提高了模型训练的效率。
技术关键词
文本编码器 特征提取网络 样本 预测类别 图像 检测模型训练方法 解码器 交互特征 视觉特征 对象 计算机程序指令 电子设备 融合特征 可读存储介质 处理器