基于改进GAN的风机叶片表面缺陷图像生成方法研究

IMAGE GENERATION METHOD OF FAN BLADE SURFACE DEFECT BASED ON IMPROVED GAN

  • 摘要: 在计算机视觉广泛应用于缺陷检测领域的背景下,针对风力发电机叶片的缺陷检测这一新应用场景,提出一种基于改进Cycle-GAN的数据增强方法,以生成高质量风机叶片表面细小裂纹图像,克服检测网络数据驱动局限性造成的应用瓶颈。该方法首先通过迁移引用和图像融合构造基本训练缺陷图像集;在Cycle-GAN基础上引入NAM注意力机制和多尺度特征融合模块,提高生成图像的质量。实验结果表明,该方法能够生成与真实缺陷相似度高的风机表面裂纹图像,且利用生成图像训练的YOLO网络平均精度(mAP@0.5)达83.7%,具有良好的工程应用价值。

     

    Abstract: Under the background that computer vision is widely-used in the field of defect detection, for the new application scenario of defect detection of wind turbine blades, a data enhancement method based on improved Cycle-GAN is proposed to generate high-quality images of small cracks on the surface of wind turbine blades so as to overcome the application bottleneck caused by the data-driven limitations of the detection network. This method constructed a basic training defect image set by transferring references and image fusion. On the basis of Cycle-GAN, NAM attention mechanism and multi-scale feature fusion module were introduced to improve the quality of generated images. The experimental results show that the method can generate fan surface crack images with high similarity to the real defects, and the average accuracy (mAP@0.5) of the YOLO network trained by the generated images reaches 83.7%, which has good engineering application value.

     

/

返回文章
返回