《Keep your Eyes on the Lane:Real-time Attention-guided Lane Detection》

本文介绍了一种基于anchor点的实时车道检测算法LaneATT,通过全局注意力机制提升检测精度。算法在Line-CNN基础上,利用RoI pooling特征的global attention,减少了计算量并保持了良好性能。实验表明,该方法在TuSimple和CULane数据集上表现出色,尤其对弯曲车道线的检测能力强。
摘要由CSDN通过智能技术生成

参考代码:LaneATT

1. 概述

导读:这篇文章提出了一种使用anchor points进行车道线检测的算法,该算法的设计源自于Line-CNN。其在Line-CNN的基础上增加了一个global attention操作(在“RoI pooling”特征基础上)使得抽取的单个anchor的RoI特征能够感知全局范围的特征,从而利用了全局信息实现车道线更好的定位。在文中还提出在训练集上统计对预先设置anchor集合进行筛选,从而减少proposal数量,可以在原本的基础上进一步减少最后的计算量,因而文章的在保持效果的同时具有较好的车道线检测性能。不过文章的车道线检测回归的目标是车道线上的点,并没有去区分车道线的实例,要确立具体的车道线实例需要采用类似启发式算法去解决。

文章的方法很大程度上源自于Line-CNN,其来源为:
Line-CNN: End-to-End Traffic Line Detection With Line Proposal Unit

在该算法中将车道线检测归纳为左右下3个方向进行RoI proposal,产生的proposal是一系列角度的点集,之后再再这些点集上进行回归,所以按照道理来讲这样的方式是可以进行一定程度扭曲程度曲线的预测的,对此文章给出的一些图例也可以看出其预测效果。Line-CNN的网络结构见下图所示:
在这里插入图片描述
可以看到它可以看作是一个“二阶段”的检测网络,其也是先在左右下3个方向上产生RoI proposal,之后将这些proposal提出再进行分类个车道线回归,下面是传统二阶段检测网络和文章提出的检测网络的对比:
在这里插入图片描述

2. 方法设计

2.1 网络结构

文章的算法在Line-CNN的基础上进行改进而来,主要的差异在使用了global attention操作,从而使得RoI proposal的特征具有了感知全局的特性。文章的网络结构见下图所示:
在这里插入图片描述

2.2 anchor机制

2.2.1 anchor数量确定

backbone产生的特征图尺寸为 F ∈ R C F ∗ H F ∗ W F F\in R^{C_F*H_F*W_F} FRCFHFWF,文章的anchor是在其左右下3个方向上进行设置的,其中左右分别6个角度变量,y轴方向上的72个切分点;下面包含15个角度变量和x轴方向的128个切分点(网络输入的图像尺寸是 360 ∗ 640 360*640 360640,所以自然x轴方向的切分相对多一些)。因而在左右下3个方向上的所有proposal总量为:
N = 2 ∗ 6 ∗ 72 + 15 ∗ 128 = 2784 N=2*6*72+15*128=2784 N=2672+15128=2784
但是这么多的数量自然带来很多的计算量,文章提出的一种解决方案是采用数据驱动的方式统计众数,之后取top-k,按照这样的思路,文章将最后取得anchor集合总数描述为 N a n c N_{anc} Nanc

2.2.2 anchor的种类与生成

将文章讲述的内容与实际的代码进行比较,发现代码中提出的anchor集合类型是有两个的,它们之间的区别在于使用的采样点数量不同,一个是使用 N p t s N_{pts} Npts(代码中描述为self.anchors),另外一个为 H F H_F HF(代码中描述为self.anchors_cut),这两种anchor在不同的阶段完成不同的任务。

对于最后用于生成检测结果使用的是self.anchors。其中采样的点数为 N p t s N_{pts} Npts,则其中在y轴方向上的采样点描述为 Y = { y i } i = 0 N p t s − 1 Y=\{y_i\}_{i=0}^{N_{pts}-1} Y={ yi}i=0Npts1,其中的一个点为 y i = i ⋅ H I N p t s − 1 y_i=i\cdot\frac{H_I}{N_{pts}-1} yi=iNpts1HI。对应的x轴方向上的集合描述为: X = { x i } i = 0 N p t s − 1 X=\{x_i\}_{i=0}^{N_{pts}-1} X={ xi}i=0Npts1,那么 x i x_i xi y i y_i

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值