Research on an Intelligent Identification Method for Wind Turbine Blade Damage Based on CBAM-BiFPN-YOLOV8

Hang Yu,Jianguo Wang,Yaxiong Han,Bin Fan,Chao Zhang

DOI: https://doi.org/10.3390/pr12010205

IF: 3.5

2024-01-18

Processes

Abstract:To address challenges in the detection of wind turbine blade damage images, characterized by complex backgrounds and multiscale feature distribution, we propose a method based on an enhanced YOLOV8 model. Our approach focuses on three key aspects: First, we enhance the extraction of small target features by integrating the CBAM attention mechanism into the backbone network. Second, the feature fusion process is refined using the Weighted Bidirectional Feature Pyramid Network (BiFPN) to replace the path aggregation network (PANet). This modification prioritizes small target features within the deep features and facilitates the fusion of multiscale features. Lastly, we improve the loss function from CIoU to EIoU, enhancing sensitivity to small targets and the perturbation resistance of bounding boxes, thereby reducing the gap between computed predictions and real values. Experimental results demonstrate that compared with the YOLOV8 model, the CBAM-BiFPN-YOLOV8 model exhibits improvements of 1.6%, 1.0%, 1.4%, and 1.1% in precision rate, recall rate, mAP@0.5, and mAP@0.5:.95, respectively. This enhanced model achieves substantial performance improvements comprehensively, demonstrating the feasibility and effectiveness of our proposed enhancements at a lower computational cost.

engineering, chemical

What problem does this paper attempt to address?

This paper attempts to solve several key problems in wind turbine blade damage detection, which mainly include:

1. **Small - target feature extraction in complex backgrounds**: In wind turbine blade damage images, due to the complex background and variable target scales, it is difficult for traditional methods to effectively extract small - target features. By introducing the Convolutional Block Attention Module (CBAM), the paper enhances the ability to extract small - target features, thereby improving the detection accuracy.

2. **Optimization of feature fusion**: The traditional Path Aggregation Network (PANet) has the problem of uneven weight distribution in the feature fusion process, which causes large - scale features to dominate in the fused output while small - scale features are ignored, possibly leading to missed detections. By introducing the Weighted Bidirectional Feature Pyramid Network (BiFPN), the paper improves the feature fusion process, enabling small - target features to be better represented in deep - layer features, thereby improving the overall detection performance.

3. **Improvement of the loss function**: The loss function of the original model has problems with height and width constraints in the shape convergence task, which may lead to large differences between the predicted box and the ground - truth box. The paper replaces the Complete Intersection over Union (CIoU) in the loss function with the Effective Intersection over Union (EIoU) to reduce the differences between the predicted box and the ground - truth box, accelerate the convergence speed of the model, and improve the prediction accuracy.

Through these improvements, the paper proposes an intelligent identification method based on CBAM - BiFPN - YOLOV8, aiming to improve the accuracy and efficiency of wind turbine blade damage detection. Experimental results show that compared with the original YOLOV8 model, the CBAM - BiFPN - YOLOV8 model has improvements of 1.6%, 1.0%, 1.4% and 1.1% in precision, recall, mAP@0.5 and mAP@0.5:0.95 respectively, demonstrating its effectiveness and feasibility in practical applications.