Efficient Automatic Driving Instance Segmentation Method Based on Detection

Chen Yanyan,Wang Hai,Cai Yingfeng,Chen Long,Li Yicheng
DOI: https://doi.org/10.19562/j.chinasae.qcgc.2023.04.002
2023-01-01
Abstract:The instance segmentation algorithm based on deep learning has achieved excellent performance in large-scale general scenarios. However, the segmentation of multi-objective instances for complex traffic scenes is still challenging, especially in the balance between high accuracy and fast inference speed, which is crucial to driving safety of intelligent vehicles. In view of this, based on the real-time algorithm Orienmask, a multihead segmentation framework is proposed based on the one-stage detection method. Specifically, the proposed framework comprises of a backbone, a feature fusion module and a multi-head mask construction module. Firstly, complete high-dimensional feature maps are obtained by adding residual structures to the backbone.Secondly, in order to generate discriminative feature representations, the feature pyramid module is reconstructed by introducing in self-calibrate convolutions and the information propagation path is improved by global attention mechanism, so as to further optimize the feature fusion module of the proposed framework. Finally, a multi-head mask construction mechanism is proposed to significantly improve the segmentation performance of different targets by refining the size distribution of instances in the traffic scenes. The proposed algorithm has been tested and validated on the open-source dataset BDD100k, and has achieved an average intersection ratio of 23.3% and 19.4%(mAP@0.5:0.95) on bounding boxes and segmentation masks, respectively. Compared with the baseline, the average index are increased by 5.2 % and 2.2 %. At the same time, the road experiment on the self-built real-vehicle platform also proves that the proposed algorithm can adapt to actual driving environments and meet the demands of real-time segmentations.
What problem does this paper attempt to address?