高级检索

    YOLO11-DWW:结构感知的木板材表面缺陷检测模型

    YOLO11-DWW: a structure-aware model for surface defect detection in wood boards

    • 摘要:
      目的 木板材表面缺陷检测是木材加工质量管控的重要环节。然而,缺陷形态多样且与木纹背景特征相似,造成细长裂纹漏检与纹理区域误检并存。为此,本文提出一种结构感知的YOLO11-DWW检测模型,旨在提高复杂纹理条件下缺陷的识别与定位精度。
      方法 以齐齐哈尔青杨类人工林采集的板材图像和木材表面公开数据子集为基础,构建包含活节、死节、裂纹节和裂纹4类缺陷的3 930幅原始图像数据集。以YOLO11n为基准,引入方向感知纹理抑制卷积(DPTSConv),增强细长裂纹的方向结构表征,并抑制规则木纹干扰;嵌入木材结构破坏感知注意力机制(WSDFAttention),强化纹理连续性破坏区域的响应;采用木材结构感知完全交并比损失函数(WSCIoU),从长宽比一致性和有效覆盖角度约束边界框回归。通过模块消融、主流检测模型对比和同类改进模块对比验证模型有效性,结合检测结果、Grad-CAM热力图及中间特征可视化分析缺陷区域响应,进一步开展公开数据集对比、复杂工况测试及不同硬件平台的部署测试。
      结果 在验证集上,YOLO11-DWW的精确率、召回率、mAP50和mAP50-95分别为0.865、0.781、0.837和0.555,较YOLO11n分别提升5.6、2.7、5.2和7.1个百分点;与YOLO11n相比,参数量减少约9.0%,浮点运算次数减少约4.8%。在Jetson Orin Nano上,YOLO11-DWW采用TensorRT FP16的推理速度为27.12帧/s,约为该模型采用PyTorch FP32时的1.79倍。
      结论 YOLO11-DWW能够增强细长缺陷的结构特征表达,降低复杂木纹背景下的误检和漏检,在保持较低模型复杂度的同时提高检测精度,可为木板材智能检测和质量分选提供技术参考。

       

      Abstract:
      Objective Surface defect detection in wood boards is an important part of quality control in wood processing. However, the diverse morphologies of defects and their similarity to background wood-grain features lead to both missed detections of slender cracks and false detections in wood-grain regions. To address these challenges, this study proposes a structure-aware YOLO11-DWW detection model to improve the accuracy of defect recognition and localization under complex texture conditions.
      Method A dataset comprising 3,930 original images covering four defect categories (live knots, dead knots, knots with cracks, and cracks) was constructed using images of wood boards sourced from poplar plantations in Qiqihar and a subset of a publicly available wood surface defect dataset. Based on YOLO11n, Directional Perception Texture Suppression Convolution (DPTSConv) was introduced to enhance the directional structural representation of slender cracks and suppress interference from regular wood-grain textures. Wood Structure Disruption-Focused Attention (WSDFAttention) was embedded to strengthen responses to regions with disrupted texture continuity. The Wood Structure-Aware Complete IoU (WSCIoU) loss was adopted to constrain bounding box regression in terms of aspect ratio consistency and effective coverage. Model effectiveness was evaluated through module ablation experiments, comparisons with mainstream object detection models, and comparisons with alternative modules of the same type. Detection results, Grad-CAM heatmaps, and intermediate feature visualizations were used to analyze responses in defect regions. Additional comparisons on a public dataset, tests under complex operating conditions, and deployment tests on different hardware platforms were conducted.
      Result On the validation set, YOLO11-DWW achieved values of 0.865, 0.781, 0.837, and 0.555 for precision, recall, mAP50, and mAP50–95, respectively, exceeding those of YOLO11n by 5.6, 2.7, 5.2, and 7.1 percentage points. Compared with YOLO11n, YOLO11-DWW reduced the parameter count and floating-point operations by approximately 9.0% and 4.8%, respectively. On the Jetson Orin Nano, YOLO11-DWW achieved an inference speed of 27.12 frames/s using TensorRT FP16, approximately 1.79 times that achieved by the same model using PyTorch FP32.
      Conclusion YOLO11-DWW enhanced the structural feature representation of slender defects and reduced false and missed detections against complex wood-grain backgrounds. It improved detection accuracy while maintaining low model complexity, providing a technical reference for intelligent wood board inspection and quality sorting.

       

    /

    返回文章
    返回