PL-UNeXt: Per-stage Edge Detail and Line Feature Guided Segmentation for Power Line Detection

Yang Cheng,Zhen Chen,Daming Liu

DOI: https://doi.org/10.1109/ICIP49359.2023.10223114

2023-09-18

Abstract:Power line detection is a critical inspection task for electricity companies and is also useful in avoiding drone obstacles. Accurately separating power lines from the surrounding area in the aerial image is still challenging due to the intricate background and low pixel ratio. In order to properly capture the guidance of the spatial edge detail prior and line features, we offer PL-UNeXt, a power line segmentation model with a booster training strategy. We design edge detail heads computing the loss in edge space to guide the lower-level detail learning and line feature heads generating auxiliary segmentation masks to supervise higher-level line feature learning. Benefited from this design, our model can reach 70.6 F1 score (+1.9%) on TTPLA and 68.41 mIoU (+5.2%) on VITL (without utilizing IR images), while preserving a real-time performance due to few inference parameters.

Computer Vision and Pattern Recognition

What problem does this paper attempt to address?

The paper aims to address the problem of power line detection, particularly the challenge of accurately separating power lines from the surrounding environment in aerial images. This task becomes especially difficult due to the complex background and the low pixel ratio of power lines in the images. To tackle these issues, the authors propose the PL-UNeXt model, a semantic segmentation model that combines edge detail guidance and line feature guidance. By introducing Edge Detail Heads to guide the encoder's lower layers in learning better edge features and Line Feature Heads to guide the higher layers in extracting relevant line feature representations, this approach enables the model to achieve an F1 score of 70.6 (an improvement of 1.9%) on the TTPLA dataset and an mIoU of 68.41 (an improvement of 5.2%) on the VITL dataset, all while maintaining real-time performance and without relying on infrared images.

Additionally, the paper conducts ablation experiments to demonstrate the effectiveness of the Edge Detail Heads and Line Feature Heads, and visualizations show how these designs help the model better focus on the shape of power lines, thereby reducing false positives and improving prediction accuracy. Overall, PL-UNeXt significantly enhances the accuracy of power line detection by leveraging prior knowledge of the edges and shapes of power lines.