Enhanced Detection of Glass Insulator Defects Using Improved Generative Modeling and Faster RCNN

Pin Ning, Jin Jin, Yuanping Xu, Chao Kong, Chaolong Zhang, Dan Tang, Jian Huang, Zhijie Xu, Tukun Li

Research output: Contribution to journalConference articlepeer-review

Abstract

The precise defect detections for glass insulators are of utmost importance to ensure their safety and functionality. Therefore, this study proposes an improved defect detection algorithm based on the optimized deep learning network. The collected glass insulator data was limited in size. Even after data augmentation, it remained insufficient for the training requirements of deep learning models. In this study, employing the improved Denoising Diffusion Probabilistic Models (DDPM) generative model to expand the glass insulator data. The glass insulator defect images generated by the improved noise-fitting network exhibit enhanced quality and high fidelity. Through manual selection and iterative experiments, a total of 1200 images were curated, constituting the glass insulator defect dataset for training and test of deep learning models. Faster RCNN was chosen as the defect detection model, and its VGG16 feature extraction network was replaced with ResNet50 to address the issue of gradient vanishing caused by excessive network stacking. Additionally, the Feature Pyramid Network (FPN) structure was introduced to enhance semantic extraction for different defect scales. The Kmeans++ algorithm was utilized to improve the proposal box generation parameters in RPN. Compared to the baseline Faster RCNN model's mAP of 72.7%, our improved version achieved a significant increase, reaching a mAP of 85.9%.
Original languageEnglish
Pages (from-to)31-36
Number of pages6
JournalProcedia CIRP
Volume129
Early online date30 Oct 2024
DOIs
Publication statusPublished - 1 Nov 2024
Event18th CIRP Conference on Computer Aided Tolerancing - Huddersfield, United Kingdom
Duration: 26 Jun 202428 Jun 2024
https://fmh.hud.ac.uk/cirp-conference/

Cite this