Toward a Cluttered Environment for Learning-Based Multi-Scale Overhead Ground Wire Recognition

摘要

In this paper, we propose a learning-based real-time method to recognize and segment an overhead ground wire (OGW) from an image, which is mainly applied to the multi-scale features in a cluttered environment. The recognition and segmentation are implemented under the framework of conditional generative adversarial nets. The generator is an end-to-end convolutional neural network (CNN) with skip connection. The discriminator is a multi-stage CNN and learns the loss function to train the generator. The OGW is recognized and shown in the downsampling visual saliency map. Thus, the location and existence of OGW are verified, which is crucial for the detection in the cluttered environment with structural lines. Detailed experiments and comparisons are performed on real-world images to demonstrate that the proposed method significantly outperforms the traditional method. Additionally, the optimized network achieves approximately 200 fps on a graphics card (GTX970) and 30 fps on an embedded platform (Jetson TX1).