摘要
本发明实施例提供了一种模型训练方法、目标检测方法、装置及电子设备,涉及计算机视觉技术领域。方法包括:获取样本图像,以及标签;针对每一样本图像,利用目标检测模型中的编码器,得到第一样本特征向量;模型还包含解码器;解码器包含至少一个部分交叉注意力模块;针对每一部分交叉注意力模块,对输入数据在通道维度进行拆分,利用该模块的自注意力层和交叉注意力层,分别对第一输入部分和第二输入部分进行处理;结合自注意力层和交叉注意力层的处理结果,得到该模块的输出数据;利用最后一个模块的输出数据得到检测结果;基于检测结果与标签计算损失值,调整模型参数,得到训练好的目标检测模型。如此,能够降低目标检测的时延。
技术关键词
注意力
前馈神经网络
样本
模块
多层感知机
解码器
数据
对象
模型训练方法
标签
特征提取单元
编码器
计算机视觉技术
模型训练装置
图像获取单元
通信接口
电子设备
通道
系统为您推荐了相关专利信息
风险预测方法
风险预测模型
风险预测装置
数据
策略
显存管理方法
计算机设备
计算机程序产品
指令
虚拟化技术
能耗管理系统
BIM技术
物理设备
定位校准
历史运行数据
考试管理系统
答题纸
智能组卷
考试管理方法
试卷管理
传输控制方法
计算机执行指令
接收随机接入响应
网络设备
信道