FDD-YOLO: A Novel Detection Model for Detecting Surface Defects in Wood

Wood surface defect detection is a critical step in wood processing and manufacturing. To address the performance degradation caused by small targets and multi-scale features in wood surface defect detection, a novel deep learning model is proposed in this study, FDD-YOLO, specifically designed for this task. In the feature extraction stage, the C2f module and the funnel attention (FA) mechanisms are integrated into the design of the C2f-FA module to enhance the model's ability to extract features of wood surface defects of various sizes. Additionally, the Dual Spatial Pyramid Pooling-Fast (DSPPF) module is developed, and the Context Self-attention Module (CSAM) is introduced to address the limitations of traditional max pooling methods, which often overlook global contextual information when extracting local features, thereby improving the detection of small-scale wood defects. In the feature fusion stage, a Dual Cross-scale Weighted Feature-fusion (DCWF) module is proposed to fuse shallow, deep, and cross-scale features through a weighted summation approach, effectively addressing the challenge of scale variation in wood surface defects. Experimental results demonstrate that the proposed FDD-YOLO model significantly improves detection performance, increasing the mAP of the baseline model YOLOv8 from 78% to 82.3%, a substantial enhancement of 4.3 percentage points. Furthermore, FDD-YOLO outperforms other mainstream defect detection models in terms of detection accuracy. The proposed model demonstrates significant potential for industrial applications by improving detection accuracy, enhancing production efficiency, and reducing material waste, thereby advancing quality control in wood processing and manufacturing.