计算机工程

作者：姜建勇, 吴云, 龙慧云, 黄自萌, 蓝林

单位：贵州大学 计算机科学与技术学院, 贵阳 550025

关键词：PD-CenterNet网络,实时检测,行人检测,样本不平衡,损失函数,特征融合

基金：国家自然科学基金（61741124）；贵州省科技计划项目（5781）。

针对传统目标检测模型不能同时兼顾检测速度和准确度的问题，提出一种新的PD-CenterNet模型。在CenterNet的基础上对网络结构和损失函数进行改进，在网络结构的上采路径中，设计基于注意力机制的特征融合模块，对低级特征和高级特性进行融合，在损失函数中通过设计α、γ、δ 3个影响因子来提高正样本与降低负样本的损失，以平衡正负样本的损失。实验结果表明，相比CenterNet模型，该模型在网络结构和损失函数上的准确度分别提高5.1%、9.81%。

来源：2021年第10期

《计算机工程》期刊编辑部

查看计算机工程杂志2021年第10期