Suppr超能文献

基于改进型CenterNet的弱感知目标检测

Weakly perceived object detection based on an improved CenterNet.

作者信息

Zhou Jing, Chen Ze, Huang Xinhan

机构信息

School of Artificial Intelligence, Jianghan University, Wuhan 430056, China.

School of Artificial Intelligence and Automation, Huazhong University of Science and Technology, Wuhan 430074, China.

出版信息

Math Biosci Eng. 2022 Sep 1;19(12):12833-12851. doi: 10.3934/mbe.2022599.

Abstract

Nowadays, object detection methods based on deep neural networks have been widely applied in autonomous driving and intelligent robot systems. However, weakly perceived objects with a small size in the complex scenes own too few features to be detected, resulting in the decrease of the detection accuracy. To improve the performance of the detection model in complex scenes, the detector of an improved CenterNet was developed via this work to enhance the feature representation of weakly perceived objects. Specifically, we replace the ResNet50 with ResNext50 as the backbone network to enhance the ability of feature extraction of the model. Then, we append the lateral connection structure and the dilated convolution to improve the feature enhancement layer of the CenterNet, leading to enriched features and enlarged receptive fields for the weakly sensed objects. Finally, we apply the attention mechanism in the detection head of the network to enhance the key information of the weakly perceived objects. To demonstrate the effectiveness, we evaluate the proposed model on the KITTI dataset and COCO dataset. Compared with the original model, the average precision of multiple categories of the improved CenterNet for the vehicles and pedestrians in the KITTI dataset increased by 5.37%, whereas the average precision of weakly perceived pedestrians increased by 9.30%. Moreover, the average precision of small objects (AP_S) of the weakly perceived small objects in the COCO dataset increase 7.4%. Experiments show that the improved CenterNet can significantly improve the average detection precision for weakly perceived objects.

摘要

如今,基于深度神经网络的目标检测方法已广泛应用于自动驾驶和智能机器人系统中。然而,在复杂场景中尺寸较小的弱感知目标拥有太少的特征以至于难以被检测到,这导致检测精度下降。为了提高检测模型在复杂场景中的性能,通过这项工作开发了一种改进的CenterNet检测器,以增强弱感知目标的特征表示。具体来说,我们用ResNext50替换ResNet50作为骨干网络,以增强模型的特征提取能力。然后,我们附加横向连接结构和空洞卷积来改进CenterNet的特征增强层,从而为弱感知目标提供丰富的特征和扩大的感受野。最后,我们在网络的检测头中应用注意力机制来增强弱感知目标的关键信息。为了证明有效性,我们在KITTI数据集和COCO数据集上评估了所提出的模型。与原始模型相比,改进后的CenterNet在KITTI数据集中车辆和行人的多类别平均精度提高了5.37%,而弱感知行人的平均精度提高了9.30%。此外,COCO数据集中弱感知小目标的小目标平均精度(AP_S)提高了7.4%。实验表明,改进后的CenterNet可以显著提高弱感知目标的平均检测精度。

文献AI研究员

20分钟写一篇综述,助力文献阅读效率提升50倍。

立即体验

用中文搜PubMed

大模型驱动的PubMed中文搜索引擎

马上搜索

文档翻译

学术文献翻译模型,支持多种主流文档格式。

立即体验