一种基于混合Transformer网络的目标检测方法、装置、电子设备及存储介质
申请号:CN202510778696
申请日期:2025-06-11
公开号:CN120599371A
公开日期:2025-09-05
类型:发明专利
摘要
本发明公开了一种基于混合Transformer网络的目标检测方法、装置、电子设备及存储介质,属于目标检测领域,所述方法包括:获取待检测图像;将待检测图像输入至目标检测模型,生成若干不同尺度的预测张量;根据若干不同尺度的预测张量,生成检测结果;其中,目标检测模型包括主干网络、颈部网络以及检测头网络;所述主干网络包括若干个依次连接的主干模块组,每一前序的主干模块组的输出为后序主干模块组的输入;至少一所述主干模块组中包括卷积视觉Transformer模块以及滑动窗口Transformer模块;通过实施本发明,能够解决现有技术存在的深层网络中小目标细微特征信息过早丢失或被稀释的问题。
技术关键词
拼接模块
卷积模块
输入端
网络
滑动窗口
输出端
检测头
采样模块
模型训练模块
图像获取模块
注意力
上采样
视觉
优化器
电子设备
风车
瓶颈
数据