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

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种模型训练方法、目标检测方法、装置及电子设备
申请号:CN202510083740
申请日期:2025-01-20
公开号:CN119904697A
公开日期:2025-04-29
类型:发明专利
摘要
本发明实施例提供了一种模型训练方法、目标检测方法、装置及电子设备,涉及计算机视觉技术领域。方法包括:获取样本图像,以及标签;针对每一样本图像,利用目标检测模型中的编码器,得到第一样本特征向量;模型还包含解码器;解码器包含至少一个部分交叉注意力模块;针对每一部分交叉注意力模块,对输入数据在通道维度进行拆分,利用该模块的自注意力层和交叉注意力层,分别对第一输入部分和第二输入部分进行处理;结合自注意力层和交叉注意力层的处理结果,得到该模块的输出数据;利用最后一个模块的输出数据得到检测结果;基于检测结果与标签计算损失值,调整模型参数,得到训练好的目标检测模型。如此,能够降低目标检测的时延。
技术关键词
注意力 前馈神经网络 样本 模块 多层感知机 解码器 数据 对象 模型训练方法 标签 特征提取单元 编码器 计算机视觉技术 模型训练装置 图像获取单元 通信接口 电子设备 通道