A deep learning based approach for automated plant disease classification using vision transformer.

Journal: Scientific reports
Published Date:

Abstract

Plant disease can diminish a considerable portion of the agricultural products on each farm. The main goal of this work is to provide visual information for the farmers to enable them to take the necessary preventive measures. A lightweight deep learning approach is proposed based on the Vision Transformer (ViT) for real-time automated plant disease classification. In addition to the ViT, the classical convolutional neural network (CNN) methods and the combination of CNN and ViT have been implemented for the plant disease classification. The models have been trained and evaluated on multiple datasets. Based on the comparison between the obtained results, it is concluded that although attention blocks increase the accuracy, they decelerate the prediction. Combining attention blocks with CNN blocks can compensate for the speed.

Authors

  • Yasamin Borhani
    Center of Excellence in Robotics and Control, Advanced Robotics & Automated Systems (ARAS), Faculty of Mechanical Engineering, K. N. Toosi University of Technology, Tehran, Iran.
  • Javad Khoramdel
    Faculty of Mechanical Engineering, Tarbiat Modares University, Tehran, Iran.
  • Esmaeil Najafi
    Center of Excellence in Robotics and Control, Advanced Robotics & Automated Systems (ARAS), Faculty of Mechanical Engineering, K. N. Toosi University of Technology, Tehran, Iran. najafi.e@kntu.ac.ir.