DE-UNet: an enhanced UNet with dual-branch attention convolution and efficient multi-scale attention aggregation for UAV lane line segmentation
摘要
Lane line detection is crucial for road environment understanding and traffic monitoring, especially in UAV-based road inspections. This study introduces an improved UNet model, DE-UNet, incorporating a dual-branch attention convolution (DBAC) module and an efficient multi-scale attention aggregation (EMSAA) module. The DBAC module enhances feature extraction by integrating parallel dilated and standard convolutions with attention mechanisms, while the EMSAA module fuses multi-level encoder features for cross-scale representation. To evaluate the model’s performance, this study created the UAV-Laneline3K dataset with more than 3000 finely annotated lane line images captured by DJI Mavic 3 T at 80–100 m and 100–120 m. Experiments demonstrate that the proposed method achieves superior performance with an F1-score of 84.11% and mIoU of 75.16% at 80–100 m altitudes, and 82.90% and 72.40% at 100–120 m, outperforming traditional methods. This research contributes to the development of robust lane line detection systems in complex road scenarios.