本平台为互联网非涉密平台,严禁处理、传输国家秘密或工作秘密

基于深度学习的烟叶等级分类及特征可视化

Tobacco leaf grading and feature visualization based on deep learning

  • 摘要: 为探索深度学习技术在烟叶图像上的特征提取效果,提出了一种基于卷积神经网络(Convolutional Neural Network,CNN)模型的烟叶等级分类方法,并对模型关注的烟叶特征进行了可视化分析。通过图像预处理得到高分辨率的局部烟叶图像,以弥补全局烟叶图像缩放后导致烟叶细节信息丢失;利用改进的CNN模型VGG-16和ResNet-50分别提取烟叶全局和局部图像特征;构建分类器对烟叶全局和局部图像的特征向量进行分类和结果融合;采用类别激活图(Class Activation Map,CAM)技术绘制模型关注烟叶特征的热力图。结果表明:提出的方法对6个等级的烟叶分级准确率达到84.71%,单张烟叶图像测试时间为17.87 ms;特征热力图显示ResNet-50模型对烟叶病斑、皱褶、主脉和纹理走势等局部特征较为敏感。该方法可为实现烟叶快速、准确分级提供支持。

     

    Abstract: To investigate the feature extraction effect of deep learning technology on tobacco leaf images, a tobacco leaf grading method based on a Convolutional Neural Network(CNN)model was proposed, and the tobacco leaf features involved in the model were visualized and analyzed. The local tobacco leaf images with high-resolution were obtained by image preprocessing to compensate for the loss of tobacco leaf details due to zooming the global tobacco leaf image. The modified CNN models VGG-16 and ResNet-50 were used to extract the features of the global and local tobacco leaf images respectively. A classifier was configured to classify and fuse the feature vectors of the global and local tobacco images. The Class Activation Map (CAM) method was used to draw the thermodynamic charts of the tobacco leaf features involved in the models. The results showed that the accuracy of the proposed method for grading tobacco leaves of six grades reached 84.71%, and the test time for a single tobacco leaf image was 17.87 ms. The thermodynamic charts of the features indicated that the ResNet-50 model was more sensitive to the local features of tobacco leaves, such as disease spot, wrinkle, main vein and texture trend. The proposed method provides support for the fast and accurate grading of tobacco leaves.

     

/

返回文章
返回