计算机与现代化 ›› 2023, Vol. 0 ›› Issue (01): 18-23.

• 图像处理 • 上一篇    下一篇

基于SE-ResNeXt的苹果叶片分类方法

  

  1. (甘肃农业大学信息科学技术学院,甘肃 兰州 730070)
  • 出版日期:2023-03-02 发布日期:2023-03-02
  • 作者简介:白旭光(1988—),男,甘肃镇原人,硕士研究生,研究方向:农业信息技术,E-mail: 1020641425@qq.com; 通信作者:刘成忠(1969—),男,甘肃天祝人,教授,硕士生导师,硕士,研究方向:人工智能,农业信息技术等,E-mail: liucz@gsau.edu.cn。
  • 基金资助:
    甘肃省高等学校创新基金资助项目(2021A-056); 甘肃省自然科学基金资助项目(20JR5RA023); 甘肃省高等学校产业支撑计划项目(2021CYZC-57); 甘肃农业大学研究生重点课程建设项目(GSAU-ZDKC-2006)

Classification Method of Small Sample Apple Leaves Based on SE-ResNeXt

  1. (College of Information Science and Technology, Gansu Agricultural University, Lanzhou 730070, China)
  • Online:2023-03-02 Published:2023-03-02

摘要: 基于现有深度学习技术,采用基于残差神经网络ResNet的变体SE-ResNeXt,构建可以自动进行苹果品种分类的卷积神经网络模型,并基于迁移学习方法训练模型。数据来源于甘肃省静宁县苹果产业基地拍摄的20类苹果叶片图像,其中每类苹果叶的图片数据量为50幅,合计1000幅。在该数据集上,对ResNet50、ResNet101、SE-ResNet50、SE-ResNet101、SE-ResNeXt50、SE-ResNeXt101这6个模型进行对比实验。结果表明,SE-ResNeXt101的结果优于其它对比模型,最高准确率达到97.5%,单张图片推断时间仅0.125 s。本文方法为今后苹果种植过程中高效、准确地识别苹果品种提供了一种手段,对辅助农技科研与苹果种植具有较大的帮助作用。

关键词: 深度学习, 残差神经网络, 迁移学习, 苹果分类, 苹果叶片图像

Abstract: Based on the existing deep learning technology, this study adopts the variant SE-ResNeXt based on residual neural network to construct a convolutional neural network model wich can automatically classify apple varieties and train the model based on transfer learning method. The data is taken from 20 types apple leaves images taken at the Apple Industry Base in Jingning County, Gansu Province. There are 50 pictures of each type of apple leaves, 1000 pictures in total. On this dataset, six models, likes ResNet50,ResNet101,SE-ResNet50,SE-ResNet101,SE-ResNeXt50 and SE-ResNeXt101, are carried out comparison experiments. The results show that SE-ResNeXt101 outperforms other comparison models, with the highest accuracy rate of 97.5% and the inference time of single image only 0.125 s. The method proposed in this paper provides a mean for identifying apple varieties efficiently and accurately in the future, and can be a great help for assisting agricultural research and apple planting.

Key words: deep learning, residual neural network, transfer learning, apple classification, apple leaf image