Improved MobileViT network for tomato leaf disease identification
DOI:
CSTR:
Author:
Affiliation:

1.School of Electronic and Information Engineering, Nanjing University of Information Science and Technology, Nanjing 210044, China; 2.Jiangsu Collaborative Innovation Center of Atmospheric Environment and Equipment Technology, Nanjing University of Information Science and Technology,Nanjing 210044, China

Clc Number:

TP391.4

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    To solve the problem of poor classification effect of convolution neural network on tomato leaf type diseases, a tomato disease identification method based on MobileViT lightweight network was proposed in this paper. Firstly, feature fusion of input and global representation is deleted and local and global representation are fused to make local representation more closely related to global representation. Secondly, in order to avoid the large increase of parameters and FLOPS when the model is scaled, the 3×3 convolution layer is replaced by 1×1 convolution layer in the fusion block. Then, the residual structure of input and fusion blocks is added to optimize the deeper level in the network model. Finally, the model accuracy is further improved by replacing the ReLU6 activation function with the H-Swish activation function. The experimental results showed that the improved MobileViT model can well recognize tomato diseases, with an average recognition accuracy of 99.16%. Compared with other convolution neural network models, it has higher recognition accuracy.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:
  • Revised:
  • Adopted:
  • Online: January 18,2024
  • Published:
Article QR Code