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.

Full Text
Published version (Free)

Talk to us

Join us for a 30 min session where you can share your feedback and ask us any queries you have

Schedule a call