APSIPA Transactions on Signal and Information Processing > Vol 13 > Issue 1

Lightweight Plank Grading Based on a Multichannel Spatial Attention Mechanism

Ziqiang Li, University of Jinan, China, Huixuan Wang, China Mobile Group Shandong Co., Ltd., China, Na Lv, University of Jinan, China, Yihong Guo, Mstar Machinery Co., Ltd., China, Yan Zhang, Shandong Jianzhu University, China
 
Suggested Citation
Ziqiang Li, Huixuan Wang, Na Lv, Yihong Guo and Yan Zhang (2024), "Lightweight Plank Grading Based on a Multichannel Spatial Attention Mechanism", APSIPA Transactions on Signal and Information Processing: Vol. 13: No. 1, e27. http://dx.doi.org/10.1561/116.20240050

Publication Date: 31 Oct 2024
© 2024 Z. Li, H. Wang, N. Lv, Y. Guo and Y. Zhang
 
Subjects
Deep learning,  Classification and prediction,  Design and analysis of algorithms,  Image and video processing,  Feature detection and selection
 
Keywords
Wood plank image gradingreal-time classificationfeature fusionlightweight neural networkconvolutional neural network
 

Share

Open Access

This is published under the terms of CC BY-NC.

Downloaded: 57 times

In this article:
Introduction 
Related Works 
Proposed Methodology 
Experiment 
Conclusion 
References 

Abstract

The grading and sorting process of the wood planks is a critical stage within the production line. However, in the real world, many factories still rely on humans to perform this task manually. This method is not only inefficient, but also time-consuming and laborintensive. To solve this problem, a lightweight wood board image classification algorithm based on a multichannel spatial attention mechanism is proposed in this paper, which can be used for the real-time classification of wood planks on the production line. This method is used to classify freshly rotated cut wood planks based on defects such as damage and voids on the production line. Specifically, the received images of the wood planks were processed by a feature extraction module to effectively separate the interfering background from the foreground of the wood planks. After fusing the edge information map with the foreground image of the wood planks, a multichannel convolutional neural network with spatial and channel attention ability was used to learn the features for correctly grading the wood planks. Experimental results show that the proposed method is superior to traditional methods and some existing deep learning algorithms in terms of performance and benefits.

DOI:10.1561/116.20240050