PISCFF-LNet: A Method for Autonomous Flight of UAVs Based on Lightweight Road Extraction

Currently, autonomous flight control for unmanned aerial vehicles (UAVs) has become increasingly critical in remote-sensing applications, such as high-resolution data acquisition and road disease detection. However, this task also faces significant challenges, particularly the weak GNSS signals in flight areas and the complex flight environment. Furthermore, many existing autonomous-flight-control algorithms for UAVs are computationally demanding, which limits their deployment on embedded devices with constrained memory and processing power, thereby affecting both operational efficiency and the safety of UAV missions. To address these issues, we propose PISCFF-LNet, a lightweight road-extraction network that integrates prior knowledge and spatial contextual features. The network employs a dual-branch encoder architecture to separately extract spatial and contextual features, thus obtaining multi-dimensional feature representations. In addition, to enhance the integration of different features and improve the overall feature representation, we also introduce a feature-fusion module. To further enhance UAV performance, we introduce an improved ray-based eight neighborhood algorithm (RENA), which efficiently extracts road-edge information with a remarkably low latency of just 7 ms, providing accurate flight guidance and reducing misidentification. To provide a comprehensive evaluation of the model’s performance, we have developed a new drone remote-sensing road-semantic-segmentation dataset, DRS Road, which includes approximately 2600 ultra-high-resolution remote-sensing images across six scene categories. The experimental results demonstrate that PISCFF-LNet achieves improvements of 1.06% in Intersection over Union (IoU) and 0.83% in F1-Score on the DeepGlobe Road dataset, and 1.03% in IoU and 0.57% in F1-Score on the DRS Road dataset, compared to existing methods. Finally, we applied the algorithm to a UAV, using a PID-based flight-control algorithm. The results show that drones employing our algorithm exhibit superior flight performance in both simulated and real-world environments.