School of Information and Electrical Engineering, Shandong Jianzhu University, Jinan 250101, China.
Research Institute of Wood Industry, Chinese Academy of Forestry, Beijing 100091, China.
Sensors (Basel). 2022 Oct 12;22(20):7733. doi: 10.3390/s22207733.
Particleboard surface defects have a significant impact on product quality. A surface defect detection method is essential to enhancing the quality of particleboard because the conventional defect detection method has low accuracy and efficiency. This paper proposes a YOLO v5-Seg-Lab-4 (You Only Look Once v5 Segmentation-Lab-4) model based on deep learning. The model integrates object detection and semantic segmentation, which ensures real-time performance and improves the detection accuracy of the model. Firstly, YOLO v5s is used as the object detection network, and it is added into the SELayer module to improve the adaptability of the model to receptive field. Then, the Seg-Lab v3+ model is designed on the basis of DeepLab v3+. In this model, the object detection network is utilized as the backbone network of feature extraction, and the expansion rate of atrus convolution is reduced to the computational complexity of the model. The channel attention mechanism is added onto the feature fusion module, for the purpose of enhancing the feature characterization capabilities of the network algorithm as well as realizing the rapid and accurate detection of lightweight networks and small objects. Experimental results indicate that the proposed YOLO v5-Seg-Lab-4 model has mAP (Mean Average Precision) and mIoU (Mean Intersection over Union) of 93.20% and 76.63%, with a recognition efficiency of 56.02 fps. Finally, a case study of the Huizhou particleboard factory inspection is carried out to demonstrate the tiny detection accuracy and real-time performance of this proposed method, and the missed detection rate of surface defects of particleboard is less than 1.8%.
刨花板表面缺陷对产品质量有重大影响。由于传统的缺陷检测方法精度和效率低,因此需要一种表面缺陷检测方法来提高刨花板的质量。本文提出了一种基于深度学习的 YOLO v5-Seg-Lab-4(You Only Look Once v5 Segmentation-Lab-4)模型。该模型集成了目标检测和语义分割,确保了实时性能,并提高了模型的检测精度。首先,使用 YOLO v5s 作为目标检测网络,并将其添加到 SELayer 模块中,以提高模型对感受野的适应性。然后,在 DeepLab v3+的基础上设计了 Seg-Lab v3+模型。在该模型中,将目标检测网络作为特征提取的骨干网络,并将 atrus 卷积的扩展率降低到模型的计算复杂度。在特征融合模块上添加通道注意力机制,以增强网络算法的特征刻画能力,实现轻量级网络和小目标的快速准确检测。实验结果表明,所提出的 YOLO v5-Seg-Lab-4 模型具有 93.20%的 mAP(平均精度)和 76.63%的 mIoU(平均交并比),识别效率为 56.02 fps。最后,对惠州刨花板厂进行了案例研究,以证明该方法的微小检测精度和实时性能,刨花板表面缺陷的漏检率小于 1.8%。