BAGGING R-CNN: ENSEMBLE FOR OBJECT DETECTION IN COMPLEX TRAFFIC SCENES
Pengteng Li (Shenzhen University); Ying He (Shenzhen University); Dongfu Yin (Guangdong Laboratory of Artificial Intelligence and Digital Economy (SZ)); F Richard Yu (Shenzhen University); Pinhao Song (KU Leuven)
-
SPS
IEEE Members: $11.00
Non-members: $15.00
Generic object detection methods have achieved preferable results, but it is still challenging to detect objects from complicated traffic scenes like extreme illumination and adverse weather. The existing methods are not robust enough to be extended to new complex traffic scenes. To address this issue, we leverage the idea of ensemble learning for strong robustness and propose a novel Bagging Region-CNN (R-CNN) framework. Specially, we design a bagging classification branch that uses adaptive sampling to train base learners and make them different from each other. The final predictions are the ensemble of the base learners, achieving strong robustness to the challenging objects. For localizing more accurately, a progressive regression branch is proposed in which bounding boxes are continuously optimized for high quality. Extensive simulation results on TJU-DHD-traffic and Pascal VOC datasets show that our Bagging R-CNN achieves superior detection accuracy over state-of-the-art methods. The source code can be found at https://github.com/PungTeng/BaggingRCNN.