DC-SPP-YOLO: Dense Connection and Spatial Pyramid Pooling Based YOLO for Object Detection

Abstract

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 underutilization of multi-scale region features. Therefore, a dense connection (DC) and spatial pyramid pooling (SPP) based YOLO (DC-SPP-YOLO) method for ameliorating the object detection accuracy of YOLOv2 is proposed in this paper. Specifically, the dense connection of convolution layers is employed in the backbone network of YOLOv2 to 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 can learn 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 indicated 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.

arXiv comment23 pages, 9 figures, 9 tables

References in corpus (7)

• Very Deep Convolutional Networks for Large-Scale Image Recognition
• DSSD : Deconvolutional Single Shot Detector
• Training Very Deep Networks
• Recent Advances in Object Detection in the Age of Deep Convolutional Neural Networks
• Enhancement of SSD by concatenating feature maps for object detection
• Residual Features and Unified Prediction Network for Single Stage Detection
• Extend the shallow part of Single Shot MultiBox Detector via Convolutional Neural Network

Cited by in corpus (6)

• P2T: Pyramid Pooling Transformer for Scene Understanding
• A General Gaussian Heatmap Label Assignment for Arbitrary-Oriented Object Detection
• DeepSOCIAL: Social Distancing Monitoring and Infection Risk Assessment in COVID-19 Pandemic
• Aerial Images Processing for Car Detection using Convolutional Neural Networks: Comparison between Faster R-CNN and YoloV3
• Search and Rescue with Airborne Optical Sectioning
• An Autonomous Drone for Search and Rescue in Forests using Airborne Optical Sectioning