Ensemble of Two-Stage Regression Based Detectors for Accurate Vehicle Detection in Traffic Surveillance Data

摘要

The growing amount of traffic surveillance data results in an increased need for automatic detection systems to analyze the data. For this purpose, deep learning based detection frameworks like Faster R-CNN and SSD have been employed in recent years. Though the detection accuracy is clearly improved compared to conventional detection methods, there exists large potential for further improvements especially in case of adverse weather conditions. In this paper, we employ the RefineDet detection framework as it combines advantages of several detection frameworks including Faster R-CNN and SSD. We use an ensemble of two detectors with different base networks to generate detections that are more robust. For this, SENets - the winner of the ImageNet2017 classification challenge - are used in addition to ResNet-50. To account for small vehicles in the background and strong variation in vehicle scale, we apply multi-scale testing. Our proposed detector achieves top-performing results on the UA-DETRAC dataset especially in case of rainy and nighttime scenarios.