YOLOX-EC: A Pedestrian and Vehicle Detection Algorithm in Automatic Driving Scenes

Di Guo,Jie Hu,Fuwu Yan
DOI: https://doi.org/10.1145/3573428.3573582
2023-01-01
Abstract:It is particularly significant that autonomous vehicles are able to accurately detect the dynamic information of pedestrians and vehicles in complex and changeable traffic scenes. However, the target detection is difficult in the automatic driving scene because the target scale varies drastically and there is mutual occlusion between targets. In order to solve the issues, we propose a pedestrian and vehicle detection algorithm that is named YOLOX-EC. Firstly, we design the Efficient Attention Module (EAM) according to the ideas of Efficient Channel Attention Network (ECANet) and Convolutional Block Attention Module (CBAM), and introduce it into backbone of YOLOX to enhance the feature extraction ability of network; Secondly, we use Content-aware reassembly of features(Carafe) to enlarge the receptive field of upsampling features and enrich the semantic information of the features; Finally, we change the bounding box regression loss to Efficient Intersection over Union Loss (EIoU Loss) for improving the regression accuracy. The experimental results on KITTI dataset show that compared with YOLOX, mean Average Precision (mAP) of the YOLOX-EC is increased by 2.8%, and the reasoning speed still keeps the real-time requirement of automatic driving platform.
What problem does this paper attempt to address?