Real-Time Classification of Rubber Wood Boards Using an SSR-Based CNN

The classification of wood types plays an important role in many fields, especially in construction industry and furniture manufacturing. In order to manufacture rubber wood furniture with highly uniform color and texture, wood boards of different colors and textures should be classified elaborately. Many traditional methods have been applied in wood classification relying on extracting features using handcrafted descriptors designed by experienced experts, but it is not easy to construct robust features in various conditions. In this article, we present a split-shuffle-residual (SSR)-based CNN that can learn features automatically from wood images for real-time classification of rubber wood boards. Specifically, we introduce an SSR module that combines channel split and shuffle operations with residual structure to reduce the computation cost while maintaining high classification accuracy. In each module, the input is split into two low-dimensional branches, and the channel shuffle operation is used to enable the information communication between the input and the two separated branches, which is regarded as the feature reuse that enlarges network capacity without increasing complexity. The comprehensive experiments demonstrate that our algorithm outperforms other traditional classification methods and the state-of-the-art deep learning classification networks, yielding an accuracy of 94.86%. Furthermore, the analysis of running time indicates that the SSR-based CNN can be employed for wood classification in real time, which takes only 26.55 ms to handle a single image.

[1]  Danilo P. Mandic,et al.  Biometrics from Brain Electrical Activity: A Machine Learning Approach , 2007, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[2]  Ali Farhadi,et al.  XNOR-Net: ImageNet Classification Using Binary Convolutional Neural Networks , 2016, ECCV.

[3]  Song Wang,et al.  Automatically Learning Semantic Features for Defect Prediction , 2016, 2016 IEEE/ACM 38th International Conference on Software Engineering (ICSE).

[4]  Ming Yang,et al.  Classification of Alzheimer’s Disease Based on Eight-Layer Convolutional Neural Network with Leaky Rectified Linear Unit and Max Pooling , 2018, Journal of Medical Systems.

[5]  Xiangyu Zhang,et al.  ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices , 2017, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[6]  Gang Wang,et al.  Deep Learning-Based Classification of Hyperspectral Data , 2014, IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.

[7]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[8]  Michele Dalponte,et al.  Tree Species Classification in Boreal Forests With Hyperspectral Data , 2013, IEEE Transactions on Geoscience and Remote Sensing.

[9]  Yu Xue,et al.  Text classification based on deep belief network and softmax regression , 2016, Neural Computing and Applications.

[10]  Min Liu,et al.  DeepSeed Local Graph Matching for Densely Packed Cells Tracking , 2021, IEEE/ACM Transactions on Computational Biology and Bioinformatics.

[11]  Jürgen Schmidhuber,et al.  LSTM: A Search Space Odyssey , 2015, IEEE Transactions on Neural Networks and Learning Systems.

[12]  Min Liu,et al.  Classification of nematode image stacks by an information fusion based multilinear approach , 2017, Pattern Recognit. Lett..

[13]  P. S. Hiremath,et al.  Multiresolution LDBP descriptors for texture classification using anisotropic diffusion with an application to wood texture analysis , 2017, Pattern Recognit. Lett..

[14]  Nikolaos Grammalidis,et al.  Wood species recognition through multidimensional texture analysis , 2018, Comput. Electron. Agric..

[15]  Paul F. Whelan,et al.  Using filter banks in Convolutional Neural Networks for texture classification , 2016, Pattern Recognit. Lett..

[16]  Sergey Ioffe,et al.  Rethinking the Inception Architecture for Computer Vision , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[17]  Amit K. Roy-Chowdhury,et al.  Multilinear feature extraction and classification of multi-focal images, with applications in nematode taxonomy , 2010, 2010 IEEE Computer Society Conference on Computer Vision and Pattern Recognition.

[18]  Xiangyu Zhang,et al.  ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design , 2018, ECCV.

[19]  Li Ma,et al.  Local Manifold Learning-Based $k$ -Nearest-Neighbor for Hyperspectral Image Classification , 2010, IEEE Transactions on Geoscience and Remote Sensing.

[20]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.

[21]  Khurram Kamal,et al.  Wood defects classification using laws texture energy measures and supervised learning approach , 2017, Adv. Eng. Informatics.

[22]  Xi Vincent Wang,et al.  A Multilevel Information Fusion-Based Deep Learning Method for Vision-Based Defect Recognition , 2020, IEEE Transactions on Instrumentation and Measurement.

[23]  Laurie A. Chisholm,et al.  Classification of Australian Native Forest Species Using Hyperspectral Remote Sensing and Machine-Learning Classification Algorithms , 2014, IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.

[24]  Bo Chen,et al.  MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications , 2017, ArXiv.

[25]  Robert X. Gao,et al.  DCNN-Based Multi-Signal Induction Motor Fault Diagnosis , 2020, IEEE Transactions on Instrumentation and Measurement.

[26]  Oscar Déniz-Suárez,et al.  Face recognition using Histograms of Oriented Gradients , 2011, Pattern Recognit. Lett..

[27]  Yiming Zhang,et al.  Diagonalwise Refactorization: An Efficient Training Method for Depthwise Convolutions , 2018, 2018 International Joint Conference on Neural Networks (IJCNN).

[28]  Carola-Bibiane Schönlieb,et al.  Individual Tree Species Classification From Airborne Multisensor Imagery Using Robust PCA , 2016, IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.

[29]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[30]  Lianwen Jin,et al.  A New CNN-Based Method for Multi-Directional Car License Plate Detection , 2018, IEEE Transactions on Intelligent Transportation Systems.

[31]  Luiz Eduardo Soares de Oliveira,et al.  Forest Species Recognition Using Deep Convolutional Neural Networks , 2014, 2014 22nd International Conference on Pattern Recognition.

[32]  N. Kanopoulos,et al.  Design of an image edge detection filter using the Sobel operator , 1988, IEEE J. Solid State Circuits.

[33]  Yu Wang,et al.  Lednet: A Lightweight Encoder-Decoder Network for Real-Time Semantic Segmentation , 2019, 2019 IEEE International Conference on Image Processing (ICIP).

[34]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[35]  M. L. Dewal,et al.  Multiresolution local binary pattern variants based texture feature extraction techniques for efficient classification of microscopic images of hardwood species , 2015, Appl. Soft Comput..

[36]  Martín Abadi,et al.  TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems , 2016, ArXiv.

[37]  Yaonan Wang,et al.  DeepBranch: Deep Neural Networks for Branch Point Detection in Biomedical Images , 2020, IEEE Transactions on Medical Imaging.

[38]  Min Liu,et al.  CNN-based two-stage cell segmentation improves plant cell tracking , 2019, Pattern Recognit. Lett..

[39]  Mengmeng Zhang,et al.  Medical Hyperspectral Image Classification Based on End-to-End Fusion Deep Neural Network , 2019, IEEE Transactions on Instrumentation and Measurement.

[40]  François Chollet,et al.  Xception: Deep Learning with Depthwise Separable Convolutions , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[41]  Kaiming He,et al.  Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks , 2015, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[42]  Dejian Li,et al.  Wood Defect Classification Based on Two-Dimensional Histogram Constituted by LBP and Local Binary Differential Excitation Pattern , 2019, IEEE Access.

[43]  Mark Sandler,et al.  MobileNetV2: Inverted Residuals and Linear Bottlenecks , 2018, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[44]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[45]  Hanchuan Peng,et al.  3D neuron tip detection in volumetric microscopy images using an adaptive ray-shooting model , 2018, Pattern Recognit..

[46]  Juho Kim,et al.  Kapre: On-GPU Audio Preprocessing Layers for a Quick Implementation of Deep Neural Network Models with Keras , 2017, ArXiv.