Research on Wood Defect Detection Algorithms Based on Improved Neural

Yuan Zhang; Xuewen Ding; Xiaokai Jiang; Shaosai Wang

doi:10.54097/3e63dg55

Authors

Yuan Zhang
Xuewen Ding
Xiaokai Jiang
Shaosai Wang

DOI:

https://doi.org/10.54097/3e63dg55

Keywords:

Wood Surface Defect Detection, Neural Network Improvement, SI, C3K2_RepVGG, Adaptive Downsampling (Adown), C2BRA

Abstract

To address the challenges of complex texture interference and the difficulty of detecting small-scale defects on wood surfaces, this study proposes an improved neural network-based detection algorithm, YOLOv11-RABRA. First, a Single Class Improve (SI) strategy is employed to selectively augment minority defect categories, thereby alleviating class imbalance and enhancing the model’s learning capability for rare defects. On this basis, a C3K2_RepVGG module is designed by integrating RepVGG re-parameterization, which optimizes the feature extraction and processing pipeline and strengthens defect feature representation. Furthermore, an ADown adaptive lightweight down-sampling mechanism is introduced to effectively preserve fine-grained features such as cracks, achieving a balance between accuracy and computational efficiency. Finally, a C2BRA dual-route attention module is applied to enhance defect-focused feature aggregation while suppressing interference from complex wood textures. Experimental results demonstrate that, compared with the baseline YOLOv11 model, the proposed algorithm achieves improvements of 13.7 percentage points in mAP50 and 12.8 percentage points in precision, significantly enhancing the accuracy and robustness of wood surface defect detection. In summary, the proposed method achieves advances in both detection performance and lightweight design, providing a feasible solution for high-precision detection in intelligent manufacturing and industrial quality inspection.

Downloads

Download data is not yet available.

References

[1] WANG Jun-dong, ZHAO Qian, LIU Ying, et al. Design of Laser Scanning and Detection Devices for Wood Surface Defects [J]. Forestry Machinery & Woodworking Equipment, 2016, 44(11): 24-27.

[2] SHEN Yin-xi, LIU Ying, YANG Yu-tu. Surface defect detection of solid wood board based on improved YOLOv5 algorithm [J]. Forestry Machinery & Woodworking Equipment, 2024, 52(3): 24-29.

[3] ZHOU Zhu, YIN Jianxin, ZHOU Suyin, ZHOU Houkui. Knot detection on coniferous wood surfaces based on near infrared spectroscopy [J]. Journal of Zhejiang A&F University, 2017, 34(3): 520-527.

[4] Qi D W, Zhang P, Zhang X F, et al. Edge detection of wood defects in X-ray wood image using neural network and mathematical morphology [C]// Technical Committee on Control Theory, Chinese Association of Automation. Proceedings of the 29th Chinese Control Conference. Beijing: China Machine Press, 2010: 2450-2455.

[5] Li C, Zhang Y, Tu W, et al. Soft measurement of wood defects based on LDA feature fusion and compressed sensor images[J]. Journal of Forestry Research, 2017, 28(6): 1285-1292.

[6] XIE Le; HENG Xi-dan;LIU Yang;JIANG Qi-long;LIU Dong. Transformer fault diagnosis based on linear discriminant analysisand step-by-step machine learning [J]. Journal of Zhejiang University: Engineering Science, 2020, 54(11): 2266-2272.

[7] Yuce B, Mastrocinque E, Packianather M S, et al. Neural network design and feature selection using principal component analysis and Taguchi method for identifying wood veneer defects[J]. Production & Manufacturing Research, 2014, 2(1): 291-308.

[8] Luo Jiancheng, Zhou Chenghu, Yang Yan. ANN Remote Sensing Classification Model and Its Integration Approach with Geo-knowledge [J]. NATIONAL REMOTE SENSING BULLETIN, 2001, 5(2): 122-129.

[9] ZHAO Shuxuan;ZHANG Jie;WANG Junliang;XU Chuqiao. Fabric Defect Detection Algorithm Based on Two-stage Deep Transfer Learning [J]. Journal of Mechanical Engineering, 2021, 57(17): 86-97.

[10] WANG Ning;ZHI Min. Review of One-Stage Universal Object Detection Algorithms in Deep Learning [J]. Journal of Frontiers of Computer Science and Technology, 2025, 19(5): 1115-1140.

[11] Shanshan Xu, Yingan Liu, Sheng Xu.Wood defects recognition based on the convolutional neural network [J]. Journal of Shandong University (Engineering Science), 2013, 43(2): 23-28.

[12] Xia B, Luo H, Shi S. Improved Faster R-CNN based surface defect detection algorithm for plates[J]. Computational Intelligence and Neuroscience, 2022: 1-11.

[13] LI Haiying; SUN Yue; ZHANG Xiao;SONG Jiancheng. Fault Diagnosis of Vacuum Contactor Based on Modal Time-frequency Diagram and ResNet50 [J]. High Voltage Engineering, 2023, 49(5): 1831-1840. DOI:10.13336/j.1003-6520. hve.20221305.

[14] ZHAO Fei;ZHANG Wenkai;YAN Zhiyuan;YU Hongfeng; DIAO Wenhui. Multi-feature Map Pyramid Fusion Deep Network for Semantic Segmentation on Remote Sensing Data [J]. Journal of Electronics & Information Technology, 2019, 41 (10): 2525-2531.

[15] Li D, Xie W, Wang B, et al. Data augmentation and layered deformable Mask R-CNN-based detection of wood defects[J]. IEEE Access, 2021, 9: 108162-108174.

[16] Yu Jiayong, Li Feng, Xue Xiankai, et al. Intelligent Identification of Bridge Structural Cracks Based on Unmanned Aerial Vehicle and Mask R-CNN [J]. China Journal of Highway and Transport, 2021, 34(12): 80-90. DOI:10. 19721/ j. cnki.1001-7372.2021.12.007.

[17] Hu Zhongkang. Research on Optimization System of Solid Wood Based on Machine Vision [D]. Nanjing: Nanjing Forestry University, 2019.

[18] CHEN Dehai; SUN Shiru;WANG Yuzhao;LEI Zhijun. Research on small target detection based on improved SSD algorithm [J]. Transducer and Microsystem Technologies, 2023, 42 (3): 65-68+72.

[19] Jia Haonan, Xu Huadong, Wang Lihai, et al. Quantitative identification of surface defects in wood paneling based on improved YOLOv5 [J]. Journal of Beijing Forestry University, 2023, 45(4): 147-155.

[20] QIU Tianheng;WANG Ling;WANG Peng;BAI Yan’e. Research on Object Detection Algorithm Based on Improved YOLOv5 [J]. Computer Engineering and Applications, 2022, 58 (13): 63-73.

[21] Han S, Jiang X, Wu Z. An improved YOLOv5 algorithm for wood defect detection based on attention[J]. IEEE Access, 2023, 11: 71800-71810.

[22] Zhang Zilei. Research on Defect Detection Algorithm of Sawn Timber Based on Deep Learning [D]. Harbin: Harbin University of Science and Technology, 2023.

[23] GENG Yanli;LIN Yanbo;FU Yanfang;YANG Shucai. Study on pigs target detection algorithm based on Transformer and adaptively spatial feature fusion [J]. Journal of Northeast Agricultural University, 2023, 54(1): 88-96.

[24] Yang Feifan, Xu Weicheng, Chen Shengde, et al.Rice disease image classification based on 1 fusion of Focal Loss and typical convolutional neural network structure [J]. Jiangsu Agricultural Sciences, 2023, 51(14): 198-204. DOI: 10. 15889/ j. issn. 1002-1302.2023.14.027.

[25] Cheng R J, Yang Y, Li L W, et al. Lightweight residual network based on depthwise separable convolution for hyperspectral image classification [J]. Acta Optica Sinica, 2023, 43 (12): 311-320.