首页|基于改进CenterNet的行车目标检测算法研究

基于改进CenterNet的行车目标检测算法研究

扫码查看
在多目标检测中,当目标发生移动时会导致目标形态变化或目标发生遮挡,影响目标检测精度,因此,针对在自动驾驶领域中行驶环境复杂以及传统目标检测算法依赖大量预设置先验框、泛化能力差、检测精度低等问题,提出了一种改进的CenterNet目标检测算法.选取DLA-34为主干特征提取网络,并引入可自适应确定卷积核且能跨通道交互的轻量级模块ECA-Net,实现CenterNet改进.在kitti数据集上的实验结果显示,改进后的CenterNet相比原网络的AP在car类别上提升了 1.39%,在pedestrian类别上提升了 11.16%,在cyclist类别上提升了 19.45%,与Yolov3网络相比,改进后的CenterNet在不同类别目标检测精度上也有着明显提升.
Research on Driving Target Detection Algorithm Based on Improved Centernet
In multi-target detection,when the target moves,it can cause changes in its shape or occlusion,which af-fects the accuracy of target detection.Therefore,to address the complex driving environment in the field of au-tonomous driving and the dependence of traditional object detection algorithms on a large number of pre-set prior boxes,poor generalization ability,and low detection accuracy,an improved CenterNet object detection algorithm is pro-posed.DLA-34 is selected as the backbone feature extraction network,and a lightweight module ECA-Net that can adaptively determine convolution kernels and interact across channels is introduced to achieve CenterNet improvement.The experimental results on the kitti dataset showed that the improved CenterNet improved the AP of the original network by 1.39%in the car category,11.16%in the pedestrian category,and 19.45%in the cyclist category.Com-pared with the Yolov3 network,the improved CenterNet also showed significant improvements in object detection ac-curacy in different categories.

autonomous drivingECA-NetDLA-34object detectionCenterNet

刘凯、宋小军

展开 >

上海电力大学 电子与信息工程学院,上海 200120

自动驾驶 ECA-Net DLA-34 目标检测 CenterNet

2024

自动化与仪表
天津市工业自动化仪表研究所 天津市自动化学会

自动化与仪表

CSTPCD
影响因子:0.548
ISSN:1001-9944
年,卷(期):2024.39(10)