基于多尺度时序采样的多任务感知网络

Multi Task Perception Network Based on Multi-Scale Temporal Sampling

  • 摘要: 针对时序特征融合不充分,遮挡及远距离目标难以可靠感知的问题,提出了一种联合时序多尺度鸟瞰视角特征的多任务感知网络. 首先,通过对深度预测概率建模,设计具备遮挡适应性的显式深度估计模块,将图像特征映射为鸟瞰视角特征,并利用深度图辅助监督;然后,为提升远距离障碍物检测效果,基于可变形注意力机制设计时序鸟瞰视角采样模块,实现时序上多尺度鸟瞰视角特征加权融合;最后,将数据增强策略拓展至多任务,并分别通过检测和分割任务头,实现三维目标检测和车道线分割. nuScenes数据集和实车实验结果证明了该方案在遮挡区域和远距离目标检测方面取得了精度提升,且推理速度可以满足实车应用要求.

     

    Abstract: Integrating temporal multi-scale bird’s eye view (BEV) features, a novel multi-task perception network was proposed to solve the problems of insufficient temporal feature fusion and the difficulty in reliably perceiving occluded or distant targets. Firstly, modeling the depth prediction probability, a module with occlusion adaptability was established to estimate visible depth, map the image features into BEV features and carry out supervision based on the depth maps. Afterwards, in order to improve the effectiveness of long-distance obstacle detection, a temporal BEV sampling module was designed based on deformable attention mechanism to make multi-scale BEV feature weighted fusion in time sequence. Finally, expanding data augmentation strategies to multi tasks, 3D object detection and lane line segmentation were achieved according to corresponding task heads separately. The results from nuScenes dataset and real-vehicle experiment show that this solution can improve the accuracy in detecting occluded areas and distant targets, and the inference speed can meet the requirements of real-world applications.

     

/

返回文章
返回
Baidu
map