计算机应用

• 人工智能与仿真 •    下一篇

基于改进深度残差网络的番茄病害图像识别

方晨晨,石繁槐   

  1. 同济大学 电子与信息工程学院
  • 收稿日期:2019-08-29 发布日期:2019-08-29 出版日期:2020-05-12
  • 通讯作者: 石繁槐

Image recognition of tomato diseases based on improved deep residual network

FANG Chenchen,SHI Fanhuai   

  • Received:2019-08-29 Online:2019-08-29 Published:2020-05-12
  • Contact: SHI Fanhuai

摘要: 针对大多数深度卷积神经网络(DCNN)模型存在内存占用较多、计算资源消耗大的问题,提出一种基于改进深度残差网络(DRN)的番茄病害图像识别方法。该网络模型在传统残差神经网络的基础上,采用多尺度卷积代替原始网络结构中的单一尺度卷积,使得提取的特征更加丰富,并拓展了网络宽度,避免因网络过深引起的退化问题。为了进一步降低模型对内存占用的需求,用深度可分离卷积替换部分标准卷积,在不损失网络性能的前提下减少模型参数。为验证改进后深度残差网络模型提升番茄病害识别性能的有效性,对获得的有限番茄病害叶片图像数据集进行了样本扩充,并基于扩充后的数据集使用改进模型与几个常见深度神经网络模型进行对比实验。结果表明,改进后的深度残差网络模型可以很好地实现番茄病害的识别,平均测试识别准确率达到 98. 58%,且训练后的模型仅占 19. 0 MB,有助于将来在低性能终端上实现对番茄病害的实时诊断。

关键词: 番茄, 病害, 图像识别, 卷积神经网络, 深度残差网络

Abstract: Aiming at the problems of large memory consumption and large computational resource consumption in most Deep Convolutional Neural Network(DCNN)models,a recognition method of tomato disease images based on improved Deep Residual Network(DRN)was proposed. Based on the traditional residual neural network,the proposed network model used multi-scale convolutions to replace the single-scale convolutions in the original network architecture,which made the extracted features richer,expanded the network width and avoided the degradation problem caused by excessively deep network. In order to further reduce the memory requirement of the model,partial standard convolutions were replaced by depthwise separable convolutions,and the model parameters were reduced without degrading network performance. To verify the effectiveness of the improved deep residual network model for tomato disease recognition,the limited dataset of tomato disease leaf images was augmented. Based on the augmented dataset,the improved model was compared with several commonly used deep neural network models. The results show that the improved DRN model can identify tomato diseases well,which achieves a mean accuracy of 98. 58% on test dataset,and has the size of only 19. 0 MB. So the proposed model will help to achieve real-time diagnosis of tomato diseases on low-performance terminals in the future.

Key words: tomato, disease, image recognition, Convolutional Neural Network (CNN), Deep Residual Network (DRN)

中图分类号: