PBD-YOLO: Dual-Strategy Integration of Multi-Scale Feature Fusion and Weak Texture Enhancement for Lightweight Particleboard Surface Defect Detection

被引:0
作者
Guo, Haomeng [1 ]
Chai, Zheming [2 ,3 ]
Dai, Huize [1 ]
Yan, Lei [4 ]
Cheng, Pengle [4 ]
Yang, Jianhua [1 ]
机构
[1] Chinese Acad Forestry, Res Inst Wood Ind, Beijing 100091, Peoples R China
[2] Harbin Inst Technol, Sch Instrumentat Sci & Engn, Harbin 150006, Peoples R China
[3] Chinese Acad Sci, Shenzhen Inst Adv Technol, Shenzhen 518055, Peoples R China
[4] Beijing Forestry Univ, Sch Technol, Beijing 100083, Peoples R China
来源
APPLIED SCIENCES-BASEL | 2025年 / 15卷 / 08期
基金
国家重点研发计划;
关键词
particleboard; surface defect detection; multi-scale feature fusion; weak texture enhancement; YOLO;
D O I
10.3390/app15084343
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
Surface defect detection plays an important role in particleboard quality control. But it still faces challenges in detecting coexisting multi-scale defects and weak texture defects. To address these issues, this study proposed PBD-YOLO (Particleboard Defect-You Only Look Once), a lightweight YOLO-based algorithm with multi-scale feature fusion and weak texture enhancement capabilities. In order to improve the ability of the algorithm to extract weak texture features, the SPDDEConv (Space to Depth and Difference Enhance Convolution) module was introduced in this study, which reduced the loss of information in the down-sampling process through space-to-depth transformation and enhanced the edge information of weak texture defects through difference convolution. This approach improved the mAP (mean average precision) of weakly featured but edge-sensitive defects (such as scratches) by as much as 20.9%. In order to improve the algorithm's ability to detect multi-scale defects, this study introduced the ShareSepHead (Share Separated Head) and C2f_SAC (C2f module with Switchable Atrous Convolution) modules. ShareSepHead fused feature maps from different scales of the neck network by adding a convolutional layer with shared weights, and the C2f_SAC module adaptively fused multi-rate receptive fields through a switching mechanism. The synergistic effect of ShareSepHead and C2f_SAC improved the detection accuracy of multi-scale defects by 10.6-20.8%. The experimental results demonstrated that PBD-YOLO achieved 85.6% mAP at 50% intersection over union (IoU) and 81.4% recall, surpassing YOLOv10 by 5.5% and 13%, respectively, while reducing parameters by 11.3%. In summary, it could be better to meet the need of accurately detecting surface defects on particleboard.
引用
收藏
页数:19
相关论文
共 39 条
[1]   SLICING AIDED HYPER INFERENCE AND FINE-TUNING FOR SMALL OBJECT DETECTION [J].
Akyon, Fatih Cagatay ;
Altinuc, Sinan Onur ;
Temizel, Alptekin .
2022 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING, ICIP, 2022, :966-970
[2]   Surface defect detection methods for industrial products with imbalanced samples: A review of progress in the 2020s [J].
Bai, Dongxu ;
Li, Gongfa ;
Jiang, Du ;
Yun, Juntong ;
Tao, Bo ;
Jiang, Guozhang ;
Sun, Ying ;
Ju, Zhaojie .
ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 130
[3]  
Muhammad MB, 2021, SN Computer Science, V2, DOI [10.1007/s42979-021-00449-3, DOI 10.1007/S42979-021-00449-3, 10.1007/s42979-021-00449-3]
[4]   DEA-Net: Single Image Dehazing Based on Detail-Enhanced Convolution and Content-Guided Attention [J].
Chen, Zixuan ;
He, Zewei ;
Lu, Zhe-Ming .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2024, 33 :1002-1015
[5]   Look Closer to See Better: Recurrent Attention Convolutional Neural Network for Fine-grained Image Recognition [J].
Fu, Jianlong ;
Zheng, Heliang ;
Mei, Tao .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :4476-4484
[6]   NAS-FPN: Learning Scalable Feature Pyramid Architecture for Object Detection [J].
Ghiasi, Golnaz ;
Lin, Tsung-Yi ;
Le, Quoc V. .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :7029-7038
[7]  
Guan Y., 2024, Appl. Innov. Technol, V5, P104, DOI [10.3969/j.issn.1671-8186.2024.05.018, DOI 10.3969/J.ISSN.1671-8186.2024.05.018]
[8]  
Haoran Zhang, 2021, 2021 IEEE International Conference on Electronic Technology, Communication and Information (ICETCI), P512, DOI 10.1109/ICETCI53161.2021.9563486
[9]   LE-MSFE-DDNet: a defect detection network based on low-light enhancement and multi-scale feature extraction [J].
Hu, Weihua ;
Wang, Tao ;
Wang, Yangsai ;
Chen, ZiYang ;
Huang, Guoheng .
VISUAL COMPUTER, 2022, 38 (11) :3731-3745
[10]  
Li B., 2022, P 2022 27 INT C AUT, P1