期刊
INFORMATION SCIENCES
卷 522, 期 -, 页码 241-258出版社
ELSEVIER SCIENCE INC
DOI: 10.1016/j.ins.2020.02.067
关键词
Object detection; Convolutional neural network; YOLOv2; Dense connection; Spatial pyramid pooling
资金
- National Key RAMP
- D Program of China [2017YFF0107303]
Although the YOLOv2 method is extremely fast on object detection, its detection accuracy is restricted due to the low performance of its backbone network and the under-utilization of multi-scale region features. Therefore, a dense connection (DC) and spatial pyramid pooling (SPP) based YOLO (DC-SPP-YOLO) method is proposed in this paper for ameliorating the object detection accuracy of YOLOv2. Specifically, the backbone network of YOLOv2 adopts the dense connection of convolution layers, which strengthen the feature extraction and alleviate the vanishing-gradient problem. Moreover, an improved spatial pyramid pooling is introduced to pool and concatenate the multi-scale region features, so that the network learns the object features more comprehensively. The DC-SPP-YOLO model is established and trained based on a new loss function composed of MSE (mean square error) loss and cross-entropy loss. The experimental results indicate that the mAP (mean Average Precision) of DC-SPP-YOLO is higher than that of YOLOv2 on the PASCAL VOC datasets and the UA-DETRAC datasets. The effectiveness of DC-SPP-YOLO method proposed is demonstrated. (C) 2020 Elsevier Inc. All rights reserved.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据