| 注册
首页|期刊导航|高技术通讯|基于注意力与密集重参数化的目标检测算法

基于注意力与密集重参数化的目标检测算法

陈志旺 雷春明 吕昌昊 王婷 彭勇

高技术通讯2024,Vol.34Issue(3):233-247,15.
高技术通讯2024,Vol.34Issue(3):233-247,15.DOI:10.3772/j.issn.1002-0470.2024.03.002

基于注意力与密集重参数化的目标检测算法

Object detection algorithm based on attention and dense reparameterization

陈志旺 1雷春明 2吕昌昊 3王婷 2彭勇4

作者信息

  • 1. 燕山大学智能控制系统与智能装备教育部工程研究中心 秦皇岛 066004||燕山大学工业计算机控制工程河北省重点实验室 秦皇岛 066004
  • 2. 燕山大学智能控制系统与智能装备教育部工程研究中心 秦皇岛 066004
  • 3. 燕山大学河北省电力电子节能与传动控制重点实验室 秦皇岛 066004
  • 4. 燕山大学电气工程学院 秦皇岛 066004
  • 折叠

摘要

Abstract

This paper presents an object detection algorithm using attention and dense reparameterization to tackle chal-lenges posed by complex backgrounds and variations in object sizes,which can adversely affect detection results.The proposed algorithm consists of two key components within an efficient feature extraction network based on CSP-DarkNet:the dense reparameterization module and the coordinate and spatial attention(CASA)module.The for-mer leverages dense connections to retain shallow features while reducing network complexity through reparameter-ization structures,while the CASA module captures necessary target information.Feature fusion is performed using feature pyramid network(FPN)and path aggregation network(PAN),and upsampling is achieved through content-aware reassembly of features(CARAFE),addressing the issue of insufficient capture of rich semantic information.To enhance model capabilities,a more efficient C3-G module is introduced to obtain gradient information,and depthwise separable convolution is employed to improve computational efficiency.Lastly,the detection output is enhanced by employing a cross-domain positive-negative sample matching strategy on a larger scale,augmenting positive samples and improving detection performance.Experimental results showcase the algorithm's advance-ments,achieving mAP@0.50 scores of 57.5%and 83.0%on the MS COCO and PASCAL VOC datasets,respec-tively.

关键词

目标检测/重参数化/注意力机制/特征融合/上采样/正负样本匹配

Key words

object detection/reparameterization/attention mechanism/feature fusion/upsampling/positive and negative sample matching

引用本文复制引用

陈志旺,雷春明,吕昌昊,王婷,彭勇..基于注意力与密集重参数化的目标检测算法[J].高技术通讯,2024,34(3):233-247,15.

基金项目

国家自然科学基金(61573305)和河北省自然科学基金(F2022203038,F2019203511)资助项目. (61573305)

高技术通讯

OA北大核心CSTPCD

1002-0470

访问量0
|
下载量0
段落导航相关论文