《Obstacle Avoidance through Deep Networks based Intermediate Perception》- 论文阅读

作者在这篇文章中总共做了三件事情:
1. 首先,作者介绍了一种从图像中自动标注轨迹的方法,用其生成了一个大型训练数据集。
2. 接着,作者使用CNN估计深度和平面法向量,生成了一个粗糙的3D模型。
3. 最后,在另一个CNN模型中,利用生成的3D模型估计避障路径。

1. 建立训练集

作者使用包含室内场景以及深度图的NYUv2数据集,如果直接从如此大的数据中获取3D轨迹是一个高难度的高维度回归问题,这非常困难。所以作者考虑将轨迹简化为5类,左方,左前方,前方,右前方,右方。对于每张图片,label就是五个轨迹中的最佳轨迹。这里的轨迹使用最大色散理论方法生成,每条路径长2.5m。
那么作者是如何从深度图中计算前进方向的呢?

首先作者将深度图构建成3维空间,然后定义路径的cost function(引用自其他文章。)

J(ξ)=fobst(ξ)+ωfsmooth(ξ)fobst(ξ)=10cobs(ξ(t))||ddtξ(t)||dtfsmooth(ξ)=1210||ddtξ(t)||2dt J ( ξ ) = f o b s t ( ξ ) + ω f s m o o t h ( ξ ) f o b s t ( ξ ) = ∫ 0 1 c o b s ( ξ ( t ) ) | | d d t ξ ( t ) | | d t f s m o o t h ( ξ ) = 1 2 ∫ 0 1 | | d d t ξ ( t ) | | 2 d t

其中, fobst(ξ) f o b s t ( ξ ) 是障碍物cost function,用于惩罚距离障碍物太近。 fsmooth(ξ) f s m o o t h ( ξ ) 用于测量路径的平滑度并惩罚高偏差。其中 cobs(ξ(t))=||max(0,dmaxd(ξ(t))||2 c o b s ( ξ ( t ) ) = | | m a x ( 0 , d m a x − d ( ξ ( t ) ) | | 2 , dmax d m a x 是障碍物cost函数可以达到的最大距离,在NYU数据集中设为3.5m。 d(ξ(t)) d ( ξ ( t ) ) 是从距离图上得到的距离障碍物的距离。
这两个公式都由《Chomp: Covariant hamiltonian optimization for motion planning》定义,这部分作者提供了代码,接下来两部分,作者都未提供代码。

2. 估计深度和平面法向量

作者使用Eigen的CNN模型来估计深度和平面法向量。该CNN模型基于全图估计出一个粗略的全局输出,接着用精细尺度的局部网络来进行细化。
Depth Loss 和 Normal Loss 如下:

Ldepth=1nid2i12n2(idi)2+1nid2iLnormal=1nivi·vi L d e p t h = 1 n ∑ i d i 2 − 1 2 n 2 ( ∑ i d i ) 2 + 1 n ∑ i ∇ d i 2 L n o r m a l = − 1 n ∑ i v i · v i ∗

其中 d d 是估计深度和真实深度之间的log差距, di ∇ d i d d 的梯度值, v v v v ∗ 分别是真实的和估计的法向量。
这两个Loss function 由《Predicting Depth, Surface Normals and Semantic Labels with a Common Multi-Scale Convolutional Architecture》定义。

3. 估计轨迹

《Obstacle Avoidance through Deep Networks based Intermediate Perception》- 论文阅读_第1张图片
作者自己设计了一个CNN网络,该CNN网络从AlexNet改版而来,有两个输入,一个是估计的平面法向量,一个是估计的深度。为了保持对称性,作者将深度图变成3通道的,每个通道一样。然后深度和法向量图像分别学习卷积核,并在第四层汇总,所以最终的预测结果将会融合两种信息。在训练时,最小化标准分类交叉熵损失:

L(C,C)=1niCilog(Ci) L ( C , C ∗ ) = − 1 n ∑ i C i ∗ l o g ( C i )

其中, Ci=ezicezi,c C i = e z i ∑ c e z i , c 是softmax类别概率, z z 是CNN的卷积输出。这一步和《Predicting Depth, Surface Normals and Semantic Labels with a Common Multi-Scale Convolutional Architecture》所做的也很像

你可能感兴趣的:(小物避障)