Author:
Zhao Ziyu,Yang Xiaoxia,Zhou Yucheng,Sun Qinqian,Ge Zhedong,Liu Dongfang
Abstract
AbstractParticleboard surface defect detection technology is of great significance to the automation of particleboard detection, but the current detection technology has disadvantages such as low accuracy and poor real-time performance. Therefore, this paper proposes an improved lightweight detection method of You Only Live Once v5 (YOLOv5), namely PB-YOLOv5 (Particle Board-YOLOv5). Firstly, the gamma-ray transform method and the image difference method are combined to deal with the uneven illumination of the acquired images, so that the uneven illumination is well corrected. Secondly, Ghost Bottleneck lightweight deep convolution module is added to Backbone module and Neck module of YOLOv5 detection algorithm to reduce model volume. Thirdly, the SELayer module of attention mechanism is added into Backbone module. Finally, replace Conv in Neck module with depthwise convolution (DWConv) to compress network parameters. The experimental results show that the PB-YOLOv5 model proposed in this paper can accurately identify five types of defects on the particleboard surface: Bigshavings, SandLeakage, GlueSpot, Soft and OliPollution, and meet the real-time requirements. Specifically, recall, F1 score, mAP@.5, mAP@.5:.95 values of pB-Yolov5s model were 91.22%, 94.5%, 92.1%, 92.8% and 67.8%, respectively. The results of Soft defects were 92.8%, 97.9%, 95.3%, 99.0% and 81.7%, respectively. The detection of single image time of the model is only 0.031 s, and the weight size of the model is only 5.4 MB. Compared with the original YOLOv5s, YOLOv4, YOLOv3 and Faster RCNN, the PB-Yolov5s model has the fastest Detection of single image time. The Detection of single image time was accelerated by 34.0%, 55.1%, 64.4% and 87.9%, and the weight size of the model is compressed by 62.5%, 97.7%, 97.8% and 98.9%, respectively. The mAP value increased by 2.3%, 4.69%, 7.98% and 13.05%, respectively. The results show that the PB-YOLOV5 model proposed in this paper can realize the rapid and accurate detection of particleboard surface defects, and fully meet the requirements of lightweight embedded model.
Funder
The authors are grateful for the support of the youth fund of Shandong Natural Science Founda-tion
The Taishan Scholar Advantage Characteristic Discipline Talent Team Project of Shandong Prov-ince of China
Publisher
Springer Science and Business Media LLC
Reference21 articles.
1. Chang, L., Guo, W., Lv, B. & Zhang, Y. Development status and trends of China’s Particleboard Industry. China Wood-Based Panels 24, 1–5 (2017).
2. Wei, Z. F., Xiao, S. H., Jiang, G. Z., Wu, S. & Cheng, G. Research on surface defect detection of wood-based panels based on deep learning. China Forest Prod. Ind. 58, 21–26 (2021).
3. Way, D., Kamke, F. & Sinha, A. Influence of specimen size during accelerated weathering of wood-based structural panels. Wood Mater. Sci. Eng. 15, 1–13. https://doi.org/10.1080/17480272.2018.1459836 (2018).
4. Zhou, H. D., Ding, T., Lu, B., Jiang, X. T. & Liu, H. L. Analysis of labor usage and up grade equipment on laminating flooring. Wood Process. Mach. 26, 1–4. https://doi.org/10.13594/j.cnki.mcjgjx.2015.05.001 (2015).
5. Chacon, M. & Alons, G. Wood defects classification using A SOM/FFP approach with minimum dimension feature vector. In Advances in Neural Networks—ISNN 2006, Third International Symposium on Neural Networks, Chengdu, China, May 28—June 1 1105–1110. https://link.springer.com/chapter/10.1007%2F11760191_161 (2006).