AspectNet: Aspect-Aware Anchor-Free Detector for Autonomous Driving

Abstract

The anchor-free-based object detection method is a crucial part in an autonomous driving system because of its low computing cost. However, the under-fitting of positive samples and overfitting of negative samples affect the detection performance. An aspect-aware anchor-free detector is proposed in this paper to address this problem. Specifically, it adds an aspect prediction head at the end of the detector, which can learn different distributions of aspect ratios between other objects. The sample definition method is improved to alleviate the problem of positive and negative sample imbalance. A loss function is designed to strengthen the learning weight of the center point of the network. The validation results show that the AP50 and AP75 of the proposed method are 97.3% and 93.4% on BCTSDB, and the average accuracies of the car, pedestrian, and cyclist are 92.7%, 77.4%, and 78.2% on KITTI, respectively. The comparison results demonstrate that the proposed algorithm is better than existing anchor-free methods.