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

by   Yongxin Cao, et al.

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%


page 6

page 9

page 15


Light-YOLOv5: A Lightweight Algorithm for Improved YOLOv5 in Complex Fire Scenarios

In response to the existing object detection algorithms are applied to c...

Image deblurring based on lightweight multi-information fusion network

Recently, deep learning based image deblurring has been well developed. ...

YOLOv5s-GTB: light-weighted and improved YOLOv5s for bridge crack detection

In response to the situation that the conventional bridge crack manual d...

An Improved Lightweight YOLOv5 Model Based on Attention Mechanism for Face Mask Detection

Coronavirus 2019 has brought severe challenges to social stability and p...

Tea Chrysanthemum Detection under Unstructured Environments Using the TC-YOLO Model

Tea chrysanthemum detection at its flowering stage is one of the key com...

Improved lightweight identification of agricultural diseases based on MobileNetV3

At present, the identification of agricultural pests and diseases has th...

Detecting The Objects on The Road Using Modular Lightweight Network

This paper presents a modular lightweight network model for road objects...

Please sign up or login with your details

Forgot password? Click here to reset