改进多尺度卷积结构与高斯核的E-CenterNet算法  

E-CenterNet Algorithm with Improved Multi-Scale Convolution Structure and Gaussian Kernel

作　　者：胡松松 吴亮红[1] 张红强[1] 陈亮 周博文[1] 张侣 HU Songsong;WU Lianghong;ZHANG Hongqiang;CHEN Liang;ZHOU Bowen;ZHANG Lyu(School of Information and Electrical Engineering,Hunan University of Science and Technology,Xiangtan,Hunan 411201,China)

机构地区：[1]湖南科技大学信息与电气工程学院,湖南湘潭411201

出　　处：《计算机工程与应用》2023年第6期70-80,共11页Computer Engineering and Applications

基　　金：国家自然科学基金(61603132);国家部委基础科研计划(JCKY2019403D006);湖南省科技创新计划(2017XK2302);湖南省自然科学基金(2021JJ30280);湖南省科协科技人才托举工程(2022TJ-Q03);湖南省教育厅优秀青年项目(19B200)。

摘　　要：针对ResNet和DLA(deep layer aggregation)为主干网络的CenterNet算法特征提取能力不充分、热力图与目标真实边界框吻合度不高、关键点损失函数未充分考虑预测值对难易样本训练比重的影响等不足,提出一种改进多尺度卷积结构与高斯核的E-CenterNet算法。引入轻量型EfficientNetV2-S作为主干网络,并结合基于金字塔分割注意力网络的多尺度卷积结构对其进行改进,提高特征提取能力;对高斯核进行改进,使CenterNet产生的热力图由固定的圆形改进为随边界框宽高变化的椭圆形,增强算法对边界框宽高差异大的目标的检测能力;提出一种基于关键点预测值的关键点损失函数,提高算法对难样本的训练比例。在Pascal VOC数据集上的实验结果表明:E-CenterNet算法的mAP达到83.3%,比原始算法提升了2.6个百分点,检测性能优于CenterNet算法。As the detection algorithm,the CenterNet based on ResNet and DLA(deep layer aggregation)as the backbone doesn’t have sufficient capabilities for feature extractions,neither its heatmap accords with the target real bounding box nor the keypoint loss function fully consider the impact of the predicted value on the training proportion of difficult and easy samples.In order to solve these deficiencies,E-CenterNet with more advanced multi-scale convolution structure and Gaussian kernel is proposed.Firstly,lightweight EfficientNetV2-S is introduced as the backbone,and the multi-scale con-volution structure based on the pyramid split attention network is used to enhance the feature extraction ability.Secondly,by improving the Gaussian kernel,the heatmap generated by CenterNet is enhanced from a fixed circle to an ellipse vary-ing with the width and height of the bounding box,which enhances the detection ability of algorithm for objects with large differences in the width and height of the bounding box.Finally,the Keypoint loss function based on the predicted value is proposed in order to augment the training ratio of the algorithm E-CenterNet for difficult samples.The experimental results on the Pascal VOC indicate that the mAP of the E-CenterNet accounts for 83.3%,which is 2.6 percentage points higher than that of the CenterNet,and the detection performance of E-CenterNet is better than that of the CenterNet.

关 键 词：CenterNet 目标检测 多尺度卷积 高斯核 关键点损失函数 

分 类 号：TP183[自动化与计算机技术—控制理论与控制工程]