Abstract:In recent years, deep learning has made significant progress in wood panel defect detection. However, there are still challenges such as low detection , slow detection speed, and difficulties in deploying embedded devices on wood panel surfaces. To overcome these issues, we propose a lightweight wood panel defect detection method called YOLOv5-LW, which incorporates attention mechanisms and a feature fusion network.Firstly, to enhance the detection capability of acceptable defects, we introduce the Multi-scale Bi-directional Feature Pyramid Network (MBiFPN) as a feature fusion network. The MBiFPN reduces feature loss, enriches local and detailed features, and improves the model's detection capability for acceptable defects.Secondly, to achieve a lightweight design, we reconstruct the ShuffleNetv2 network model as the backbone network. This reconstruction reduces the number of parameters and computational requirements while maintaining performance. We also introduce the Stem Block and Spatial Pyramid Pooling Fast (SPPF) models to compensate for any accuracy loss resulting from the lightweight design, ensuring the model's detection capabilities remain intact while being computationally efficient.Thirdly, we enhance the backbone network by incorporating Efficient Channel Attention (ECA), which improves the network's focus on key information relevant to defect detection. By attending to essential features, the model becomes more proficient in accurately identifying and localizing defects.We validate the proposed method using a self-developed wood panel defect dataset.The experimental results demonstrate the effectiveness of the improved YOLOv5-LW method. Compared to the original model, our approach achieves a 92.8\% accuracy rate, reduces the number of parameters by 27.78\%, compresses computational volume by 41.25\%, improves detection inference speed by 10.16\%

A Transfer Residual Neural Network Based on ResNet-34 for Detection of Wood Knot Defects

A Transfer Residual Neural Network Based on ResNet-34 for Detection of Wood Knot Defects

A Novel Deep Convolutional Neural Network Based on ResNet-18 and Transfer Learning for Detection of Wood Knot Defects

Estimation of the Convolutional Neural Network with Attention Mechanism and Transfer Learning on Wood Knot Defect Classification

RESEARCH ON WOOD DEFECTS CLASSIFICATION BASED ON DEEP LEARNING

Detection of Knot Defects on Coniferous Wood Surface Using Near Infrared Spectroscopy and Chemometrics

BLNN: Multiscale Feature Fusion-Based Bilinear Fine-Grained Convolutional Neural Network for Image Classification of Wood Knot Defects

Data Augmentation and Layered Deformable Mask R-CNN-Based Detection of Wood Defects

Surface Detection of Solid Wood Defects Based on SSD Improved with ResNet

Wood surface defects detection based on the improved YOLOv5-C3Ghost with SimAm module

Improved ResNet-50 model for identifying defects on wood surfaces

Detection of natural wood defects with large color differences based on branched network

An accurate object detection of wood defects using an improved Faster R-CNN model

A Multi-Source Data Fusion Network for Wood Surface Broken Defect Segmentation

Lightweight wood panel defect detection method incorporating attention mechanism and feature fusion network

I2GF-Net: Inter-layer Information Guidance Feedback Networks for Wood Surface Defect Detection in Complex Texture Backgrounds

Wood defect detection based on the CWB-YOLOv8 algorithm

GBCD-YOLO: A High-Precision and Real-Time Lightweight Model for Wood Defect Detection

FDADNet: Detection of Surface Defects in Wood-Based Panels Based on Frequency Domain Transformation and Adaptive Dynamic Downsampling

An Efficient and Accurate Surface Defect Detection Method for Quality Supervision of Wood Panels

ODCA-YOLO: An Omni-Dynamic Convolution Coordinate Attention-Based YOLO for Wood Defect Detection