Paper Reading AI Learner

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

2023-03-08 07:32:01
Yang Cheng, Zhen Chen, Daming Liu

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.

Abstract (translated)

电源线路检测是电力公司的一项关键检查任务,同时也可用于避免无人机障碍。由于复杂的背景和像素比例较低,在平面图像中准确分离电源线路与周围区域仍然是一项挑战。为了正确捕捉空间边缘细节先验和线特征的指导,我们提供了PL-UNeXt,它是一个电源线路分割模型,采用增强训练策略。我们设计的边缘细节头计算边缘空间的损失,以指导较低级别细节学习,以及线特征头生成auxiliary segmentation masks,以监督更高级别的线特征学习。受益于此设计,我们的模型可以在TTPLA上获得70.6 F1得分(+1.9%),在VITL上获得68.41 mIoU(不使用红外图像) (+5.2%),同时由于 few inference parameters,仍保留实时性能。

URL

https://arxiv.org/abs/2303.04413

PDF

https://arxiv.org/pdf/2303.04413.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot