高级检索
袁鑫, 江聪世. 基于模型集成的高分辨率融合车道线检测算法[J]. 计算机辅助设计与图形学学报, 2022, 34(9): 1402-1410. DOI: 10.3724/SP.J.1089.2022.19161
引用本文: 袁鑫, 江聪世. 基于模型集成的高分辨率融合车道线检测算法[J]. 计算机辅助设计与图形学学报, 2022, 34(9): 1402-1410. DOI: 10.3724/SP.J.1089.2022.19161
Yuan Xin, Jiang Congshi. High-Resolution Fusion Lane Detection Algorithm Based on Model Ensemble[J]. Journal of Computer-Aided Design & Computer Graphics, 2022, 34(9): 1402-1410. DOI: 10.3724/SP.J.1089.2022.19161
Citation: Yuan Xin, Jiang Congshi. High-Resolution Fusion Lane Detection Algorithm Based on Model Ensemble[J]. Journal of Computer-Aided Design & Computer Graphics, 2022, 34(9): 1402-1410. DOI: 10.3724/SP.J.1089.2022.19161

基于模型集成的高分辨率融合车道线检测算法

High-Resolution Fusion Lane Detection Algorithm Based on Model Ensemble

  • 摘要: 针对无人机航拍视角下车道线形状复杂、细节特征易丢失、车道线前后景像素占比不均衡等问题,提出一种基于模型集成的高分辨率融合车道线检测算法.首先使用高分辨率融合结构和双线性插值算法改进全卷积神经网络的卷积模块和上采样模块;然后依据模型集成思想,使用改进后的模型结构作为车道线前后景语义分割模型及车道线多类别语义分割模型,用于分步骤解决车道线检测问题,并使用阈值化交叉熵损失函数和Lovasz损失函数组成联合损失函数对2种模型进行训练;最后使用局部色选区域生长算法为检测结果添补细节.实验结果表明,所提算法在自定义无人机航拍视角的15类车道线语义分割数据集中达到0.5484的平均交并比和0.9931的像素精度,在NVIDIA Tesla V100平台对分辨率为512×512的图像的检测速度达到23.08帧/s.

     

    Abstract: Lane line detection is still a challenging task due to the challenges coming from the complexity of drone aerial images such as complex lane lines,fine-grained feature,class imbalance,etc.Therefore,a lane line detection algorithm based on high-resolution fusion convolution network is proposed.Firstly,the convolution module and up sampling module of full convolution network are improved by using high-resolution fusion structure and bilinear interpolation algorithm.Then,according to the idea of model ensembling,the improved model architecture is used as the foreground-background semantic segmentation model and the multi-category semantic segmentation model,which is used to solve the problem of lane line detection step by step,and the two models are trained by the joint loss function composed of threshold cross entropy loss and Lovasz loss.Finally,the locally region-growth algorithm is used to supplement the details of the detected results.The experimental results show the algorithm achieves 0.5484 mean intersection over union and 0.9931 pixel accuracy in the customized drone aerial dataset of 15 types of lane lines,and the prediction speed of 512×512 resolution image on NVIDIA Tesla V100 reaches 23.08 frame per second.

     

/

返回文章
返回