欢迎访问林业科学,今天是

林业科学 ›› 2022, Vol. 58 ›› Issue (3): 149-158.doi: 10.11707/j.1001-7488.20220316

•   • 上一篇    下一篇

基于AlexNet优化的板材材种识别方法

杨楠楠1,白岩1,*,姜苏译2,杨春梅2,徐凯宏1   

  1. 1. 东北林业大学信息与计算机工程学院 哈尔滨 150040
    2. 东北林业大学机电工程学院 哈尔滨 150040
  • 收稿日期:2021-03-16 出版日期:2022-03-25 发布日期:2022-06-02
  • 通讯作者: 白岩
  • 基金资助:
    中央高校基本科研业务费专项资金资助项目(2572019BF03)

Recognition Method of Plate and Wood Based on ALexNet Optimaization

Nannan Yang1,Yan Bai1,*,Suyi Jiang2,Chunmei Yang2,Kaihong Xu1   

  1. 1. College of Information and Computer Engineering, Northeast Forestry University Harbin 150040
    2. College of Mechanical and Electrical Engineering, Northeast Forestry University Harbin 150040
  • Received:2021-03-16 Online:2022-03-25 Published:2022-06-02
  • Contact: Yan Bai

摘要:

目的: 基于处理后的木材端面细胞特征, 寻找合适的机器学习方法提高木材识别准确率, 以实现木材高效利用, 为珍稀木材种类判别和保护提供依据。方法: 以5种木材(臭冷杉、长白落叶松、鱼鳞云杉、鹅掌楸和凹叶厚朴)端面细胞为研究对象, 提取多种差异性图像作为数据集, 通过图像处理提取特征信息, 分别采用支持向量机(SVM)和AlexNet神经网络进行分类识别。根据木材端面细胞区分的差异性, 在AlexNet神经网络架构中加入BN算法进行优化, 设计一种更高效的板材识别方法提高木材识别准确率。结果: 将增强后的29 680张图像按7∶3划分, 分别保存在训练集和测试集文件夹中, 测试样本确定标签后均放入同一文件夹, 分别对3种分类算法进行整体批量测试, 支持向量机分类器测试集的整体识别准确率为84.67%, AlexNet神经网络测试集的整体识别准确率为88.76%, 基于BN算法优化的AlexNet神经网络测试集的整体识别准确率为91.15%, 识别效果更好。结论: 当样本量充足时, AlexNet神经网络对木材端面细胞图像的分类效果明显优于SVM分类器。基于BN算法优化的AlexNet神经网络对图像线性特征更敏感, 保留AlexNet神经网络拟合优化性的同时加快寻优速率, 可有效提高识别精度, 实现木材高精度分类。

关键词: 材种识别, 机器学习, AlexNet, SVM

Abstract:

Objective: Based on the cell characteristics of the processed image of cell on wood end faces, the search for corresponding machine learning methods can significantly increase the accuracy of identifying and realize efficient utilization and processing of wood, with the aim to provide an important basis for the identification and protection of rare wood species. Method: Using the cells on end faces of five kinds of wood(Abies nephrolepis, Larix olgensis, Picea jezoensis var.microsperma, Liriodendron chinense, Magnolia officinalis subsp. biloba) as research samples, a variety of differential images are extracted as data sets. Feature information is extracted by image processing, and support vector machine classifier(SVM) and AlexNet neural network are used for classification and recognition. According to the difference of wood end face cell distinction, BN (batch normalization) algorithm is added to the AlexNet neural network architecture for optimization, and a more efficient wood recognition method is designed. Result: The 29 680 enhanced images are divided into a 7∶3 ratio and saved in the training set and the test set folders. The test samples are labeled and put into the same folder, then conduct an overall batch test of the three classification algorithms. The results show that the overall recognition accuracy of the test set of the support vector machine classifier is 84.67%, the overall recognition accuracy of the test set of the AlexNet neural network is 88.76%, the overall recognition accuracy of the AlexNet neural network based on the BN algorithm is 91.15%. It can be seen that the recognition accuracy of the AlexNet neural network based on the BN algorithm is better. Conclusion: When the sample size is sufficient, the classification effect of the AlexNet neural network on images of cell on wood end faces is significantly better than that of the SVM classifier. The optimized AlexNet neural network based on the BN algorithm architecture is more sensitive to the linear features of the image, retains the fitting optimization of the AlexNet neural network and speeds up the optimization rate, which can effectively improve the classification accuracy and achieve high-precision wood classification.

Key words: wood species identification, machine learning, AlexNet, SVM

中图分类号: