Symbol Detection in a Multi-class Dataset Based on Single Line Diagrams using Deep Learning Models

Abstract

Single Line Diagrams (SLDs) are used in electrical power distribution systems. These diagrams are crucial to engineers during the installation, maintenance, and inspection phases. For the digital interpretation of these documents, deep learning-based object detection methods can be utilized. However, there is a lack of efforts made to digitize the SLDs using deep learning methods, which is due to the class-imbalance problem of these technical drawings. In this paper, a method to address this challenge is proposed. First, we use the latest variant of You Look Only Once (YOLO), YOLO v8 to localize and detect the symbols present in the single-line diagrams. Our experiments determine that the accuracy of symbol detection based on YOLO v8 is almost 95%, which is more satisfactory than its previous versions. Secondly, we use a synthetic dataset generated using multi-fake class generative adversarial network (MFCGAN) and create fake classes to cope with the class imbalance problem. The images generated using the GAN are then combined with the original images to create an augmented dataset, and YOLO v5 is used for the classification of the augmented dataset. The experiments reveal that the GAN model had the capability to learn properly from a small number of complex diagrams. The detection results show that the accuracy of YOLO v5 is more than 96.3%, which is higher than the YOLO v8 accuracy. After analyzing the experiment results, we might deduce that creating multiple fake classes improved the classification of engineering symbols in SLDs.