Classification of tea leaf disease using convolutional neural network approach

Abstract

Leaf diseases on tea plants affect the quality of tea. This issue must be overcome since preparing tea drinks requires high-quality tea leaves. Various automatic models for identifying disease in tea leaves have been developed; however, their performance is typically low since the extracted features are not selective enough. This work presents a classification model for tea leaf disease that distinguishes six leaf classes: algal spot, brown, blight, grey blight, helopeltis, red spot, and healthy. Deep learning using a convolutional neural network (CNN) builds an effective model for detecting tea leaf illness. The Kaggle public dataset contains 5,980 tea leaf images on a white background. Pre-processing was performed to reduce computing time, which involved shrinking and normalizing the image prior to augmentation. Augmentation techniques included rotation, shear, flip horizontal, and flip vertical. The CNN model was used to classify tea leaf disease using the MobileNetV2 backbone, Adam optimizer, and rectified linear unit (ReLU) activation function with 224×224 input data. The proposed model attained the highest performance, as evidenced by the accuracy value 0.9455.