源码详解:Vectornet源码详解_樱花的浪漫的博客-优快云博客
源码见文末
Abstract
本篇论文主要做了三个方面的事情:
-
特征工程
不同与以前的方法,将车辆行驶收集的信息用一张全局的图像来表示,论文提出用向量表示车辆行驶过程中收集的各种信息。
2. 层次图神经网络
论文构建了一种层次图神经网络,层次,即首先各个实体比如车辆、红绿灯等自身构建一个子图,然后将这些子图再连接成一层图。
3.辅助任务
为了进一步提高向量网络学习上下文特征的能力,作者提出了一种新的辅助任务,即对图随机进行mask,然后对mask部分做预测,也就是一个良好的网络结构不仅能够预测好其本身的任务,而且能够对图中缺失的结点做好预测。
1. Introduction
(1&#x