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

林业科学 ›› 2023, Vol. 59 ›› Issue (8): 133-140.doi: 10.11707/j.1001-7488.LYKX20210891

• • 上一篇    

基于卷积神经网络的家具木材图像种类识别

苗宇杰,祝诗平*,普京,李俊贤,马羚凯,黄华   

  1. 西南大学工程技术学院 重庆 400716
  • 收稿日期:2021-12-18 出版日期:2023-08-25 发布日期:2023-10-16
  • 通讯作者: 祝诗平
  • 基金资助:
    中央高校基本科研业务费专项资金项目( XDJK2019C081)

Recognition of Furniture Wood Image Species Based on Convolutional Neural Networks

Yujie Miao,Shiping Zhu*,Jing Pu,Junxian Li,Lingkai Ma,Hua Huang   

  1. College of Engineering and Technology, Southwest University Chongqing 400716
  • Received:2021-12-18 Online:2023-08-25 Published:2023-10-16
  • Contact: Shiping Zhu

摘要:

目的: 为解决日常生活中家具木材种类主要依靠人工识别存在的主观性强、效率低等问题,设计一种基于MobileNetV3卷积神经网络(CNN)的常见家具木材图像种类识别模型,有效提高木材种类识别速度和精度,为木材资源的合理利用、木材进出口贸易管理及消费者确定家具木材种类提供一种科学有效的方法。方法: 首先,采集4种木材的素材图像以及其中2种涂饰木蜡油的木材图像,共3 880张,将数据集按6∶2∶2和2∶6∶2分为训练集、验证集和测试集,基于旋转、翻转等操作将训练集数据扩充为原来的4倍。采用4种卷积神经网络和2种传统机器学习方法对素材图像建立识别模型,通过分析对比得到最优识别网络模型——MobileNetV3,对模型进行迁移学习并参数调优。将素材图像与涂饰木蜡油的木材图像对应放在一起与剩下的2种素材图像共同组成新的数据集,构建基于MobileNetV3网络的木材种类识别模型。为方便分类人员操作,减轻分类人员在实际检测中的操作难度,选用构建的木材种类识别模型,基于PyQt5搭建木材种类识别系统。结果: 与传统机器学习方法相比,卷积神经网络识别效果更佳,且迁移学习能够明显提高网络的收敛速度和分类性能。在验证集中,识别性能最佳的MobileNetV3对4种素材图像的平均识别准确率为98.13%,对混有涂饰木蜡油木材图像的识别准确率为97.25%。结论: 木材种类识别模型不仅可对素材进行识别,也能实现对涂饰木蜡油木材的快速、准确识别,为木材分类人员带来便捷的同时,也为消费者挑选实木家具并确定木材种类提供一种可靠的识别方法。

关键词: 家具木材, 木蜡油, 识别, 卷积神经网络, MobileNetV3

Abstract:

Objective: In order to solve the problems of strong subjectivity and low efficiency that the identification of furniture wood species mainly depends on manual identification in daily life, a common furniture wood species identification model based on Mobilenetv3 convolutional neural network (CNN) was designed to effectively improve the identification speed and accuracy of wood species. It provided a scientific and effective method for the rational utilization of wood resources, the management of wood import and export trade and the determination of furniture wood types by consumers. Method: Firstly, 3 880 images of four kinds of wood images coated without wood wax oil and two kinds of wood images coated with wood wax oil were collected. The data set was divided into training set, verification set and test set according to 6∶2∶2 and 2∶6∶2, and the data of the training set was expanded to four times of the original by using operations such as rotation and flip. Then four convolution neural networks and two traditional machine learning methods were used to establish the recognition model for the wood image without coated wood wax oil. Through analysis and comparison, the optimal recognition network model Mobilenetv3 was obtained, and the parameters of the model were optimized based on transfer learning. The wood species identification model based on MobileNetv3 network was constructed by putting the wood images coated without wood wax oil together with the wood images coated with wood wax oil and forming a new data set together with the remaining two kinds of wood images coated without wood wax oil. Finally, in order to simplify the operation of the classification personnel and reduce the operation difficulty of the staff in the actual detection, we selected the above wood species recognition model and built a wood species recognition system based on PyQt5. Result: Compared with traditional machine learning methods, convolutional neural network had better recognition results, and transfer learning could significantly improve the convergence speed and classification performance of the network. In the validation set, MobileNetV3 with the best recognition performance had an average recognition accuracy of 98.13% for four wood images without coated wood wax oil and 97.25% for wood images mixed with coated wood wax oil. Conclusion: In this paper, the wood species recognition model not only recognized the recognition of wood images coated without wood wax oil, but also realized the fast and accurate recognition of the painted wood wax wood image. Apart from bring convenience to wood classifiers, the wood species recognition model also could provide a reliable identification method for consumers to select solid wood furniture and determine wood species.

Key words: furniture wood, wax oil, identification, convolutional neural network, MobileNetV3

中图分类号: