GNDnet自动驾驶车辆的快速地平面估计和点云分割 实验部分
接上一篇损失函数
GNDnet的损失函数为回归损失函数和空间平滑损失的加权和
其中I表示形状为(H;W)的网格的地面真实高程图,I^表示预测高程图。超参数α和β用于平衡这两种损失。
对于高程值的回归,我们使用Huber损失。对于空间平滑度,我们最小化预测高程地图二阶梯度的L1范数。我们对相邻单元的二阶梯度范数进行惩罚,以鼓励高程值不是恒定的,而是平稳变化的:
其中
其中rx和ry分别是立面图x和y方向上的坡度。
具体的训练细节
网络结构
我们不预先训练我们的网络,所有权重都是使用均匀分布随机初始化的。支柱特征编码网络中的简化点网有一个完全连接的层,具有C=(9,64)输入输出特征。
在编码器-解码器体系结构中,所有卷积层的内核大小为3,步长和填充为1。除最后一个回归层外,每个卷积层后面是批量归一化和元素校正线性非线性(ReLU)。使用2×2窗口和跨步2执行最大池和最大取消池。使用卷积层的输入输出通道的编码器-解码器网络设计是: