Light-Weight Deep Learning Model for Accelerating the Classification of Mango-Leaf Disease

Bahar Uddin Mahmud, Abdullah Al Mamun, Md Jakir Hossen, Guan Yue Hong, Busrat Jahan


Mango leaf diseases represent a serious threat to world agriculture, necessitating prompt and accurate detection to avert catastrophic effects. In response, this study suggests a light-weight, deep learning-based method for automatically classifying mango leaf diseases. The model is based on the original DenseNet architecture, which is well known for its effectiveness in image classification tasks. Custom layers have been added over the existing layer of the original DenseNet model. The proposed model has been compared with other existing pre-trained models. Based on comparisons, the proposed model, DenseNet78, proved to be efficient even on a relatively small dataset, where the conventional model failed. The proposed model ensured generalization across regions, disease variants, and diverse datasets of mango leaves. The results demonstrate that the fine-tuned DenseNet architecture (DenseNet78), along with an ideal growth rate, modifying block size, and a number of layers, provides optimum accuracy, with 99.47% accuracy in identifying healthy mango leaves and 99.44% accuracy in detecting various mango leaf diseases. The results also demonstrate that the model is effective in accelerating the training process because of careful comparative analysis of all the available alternatives, including the most effective combination of optimizers, learning rate schedulers, and loss functions. The study's conclusion is an automated approach for diagnosing mango leaf disease using an improved and optimized DenseNet architecture (DenseNet78).


Doi: 10.28991/ESJ-2024-08-01-03

Full Text: PDF


Image Classification; Custom DenseNet; Deep Learning; Computer Vision; Leaf Disease; Disease Classification.


Mahmud, B. U., Hong, G. Y., Mamun, A. Al, Ping, E. P., & Wu, Q. (2023). Deep Learning-Based Segmentation of 3D Volumetric Image and Microstructural Analysis. Sensors, 23(5), 2640. doi:10.3390/s23052640.

Jahan, B., Mahmud, B. U., Mamun, A. A., Mujibur Rahman Majumder, Md., & Alam, M. (2021). Impact Analysis of Harassment Against Women in Bangladesh Using Machine Learning Approaches. Recent Trends in Mechatronics Towards Industry 4.0, 549–559. doi:10.1007/978-981-33-4597-3_50.

Le, V. N. T., Ahderom, S., Apopei, B., & Alameh, K. (2020). A novel method for detecting morphologically similar crops and weeds based on the combination of contour masks and filtered Local Binary Pattern operators. GigaScience, 9(3), 17. doi:10.1093/gigascience/giaa017.

Ramesh, S., Hebbar, R., Niveditha, M., Pooja, R., Shashank, N., & Vinod, P. V. (2018). Plant disease detection using machine learning. In 2018 International conference on design innovations for 3Cs compute communicate control (ICDI3C), 41-45. doi:10.1109/ICDI3C.2018.00017.

Kuricheti, G., & Supriya, P. (2019). Computer vision based turmeric leaf disease detection and classification: A step to smart agriculture. Proceedings of the International Conference on Trends in Electronics and Informatics, ICOEI 2019, 545–549. doi:10.1109/ICOEI.2019.8862706.

Saleem, R., Shah, J. H., Sharif, M., Yasmin, M., Yong, H. S., & Cha, J. (2021). Mango leaf disease recognition and classification using novel segmentation and vein pattern technique. Applied Sciences (Switzerland), 11(24), 11901. doi:10.3390/app112411901.

Srunitha, K., & Bharathi, D. (2018). Mango leaf unhealthy region detection and classification. In D. Hemanth & S. Smys (Eds.): Lecture Notes in Computational Vision and Biomechanics, 28, 422–436. doi:10.1007/978-3-319-71767-8_35.

Mia, M. R., Roy, S., Das, S. K., & Rahman, M. A. (2020). Mango leaf disease recognition using neural network and support vector machine. Iran Journal of Computer Science, 3(3), 185–193. doi:10.1007/s42044-020-00057-z.

Ullagaddi, S. B., & Raju, S. V. (2017). Disease recognition in Mango crop using modified rotational kernel transform features. 4th International Conference on Advanced Computing and Communication Systems, ICACCS 2017, Coimbatore, India. doi:10.1109/ICACCS.2017.8014610.

Ullagaddi, S. B., & Viswanadha Raju, S. (2017). An Enhanced Feature Extraction Technique for Diagnosis of Pathological Problems in Mango Crop. International Journal of Image, Graphics and Signal Processing, 9(9), 28–39. doi:10.5815/ijigsp.2017.09.04.

Abdollahi, A., & Pradhan, B. (2021). Urban vegetation mapping from aerial imagery using explainable AI (XAI). Sensors, 21(14), 4738. doi:10.3390/s21144738.

Fuentes, A., Yoon, S., Kim, S. C., & Park, D. S. (2017). A robust deep-learning-based detector for real-time tomato plant diseases and pests recognition. Sensors (Switzerland), 17(9), 2022. doi:10.3390/s17092022.

Agarwal, M., Singh, A., Arjaria, S., Sinha, A., & Gupta, S. (2020). ToLeD: Tomato Leaf Disease Detection using Convolution Neural Network. Procedia Computer Science, 167, 293–301. doi:10.1016/j.procs.2020.03.225.

Richey, B., Majumder, S., Shirvaikar, M. V., & Kehtarnavaz, N. (2020). Real-time detection of maize crop disease via a deep learning-based smartphone app. Real-Time Image Processing and Deep Learning 2020. International Society for Optics and Photonics, 10. doi:10.1117/12.2557317.

Batool, A., Hyder, S. B., Rahim, A., Waheed, N., Asghar, M. A., & Fawad. (2020). Classification and Identification of Tomato Leaf Disease Using Deep Neural Network. 2020 International Conference on Engineering and Emerging Technologies, ICEET 2020. doi:10.1109/ICEET48479.2020.9048207.

Tm, P., Pranathi, A., SaiAshritha, K., Chittaragi, N. B., & Koolagudi, S. G. (2018, August). Tomato leaf disease detection using convolutional neural networks. In 2018 Eleventh International Conference on Contemporary Computing (IC3), 1-5. doi:10.1109/IC3.2018.8530532.

Turkoglu, M., Yanikoğlu, B., & Hanbay, D. (2022). PlantDiseaseNet: convolutional neural network ensemble for plant disease and pest detection. Signal, Image and Video Processing, 16(2), 301-309. doi:10.1007/s11760-021-01909-2.

Gulavnai, S., & Patil, R. (2019). Deep Learning for Image Based Mango Leaf Disease Detection. International Journal of Recent Technology and Engineering, 8(3S3), 54–56. doi:10.35940/ijrte.c1030.1183s319.

Pham, T. N., Tran, L. Van, & Dao, S. V. T. (2020). Early Disease Classification of Mango Leaves Using Feed-Forward Neural Network and Hybrid Metaheuristic Feature Selection. IEEE Access, 8, 189960–189973. doi:10.1109/ACCESS.2020.3031914.

Singh, U. P., Chouhan, S. S., Jain, S., & Jain, S. (2019). Multilayer Convolution Neural Network for the Classification of Mango Leaves Infected by Anthracnose Disease. IEEE Access, 7, 43721–43729. doi:10.1109/ACCESS.2019.2907383.

Prabu, M., & Chelliah, B. J. (2022). Mango leaf disease identification and classification using a CNN architecture optimized by crossover-based levy flight distribution algorithm. Neural Computing and Applications, 34(9), 7311–7324. doi:10.1007/s00521-021-06726-9.

Wongsila, S., Chantrasri, P., & Sureephong, P. (2021). Machine Learning Algorithm Development for detection of Mango infected by Anthracnose Disease. 2021 Joint 6th International Conference on Digital Arts, Media and Technology with 4th ECTI Northern Section Conference on Electrical, Electronics, Computer and Telecommunication Engineering, ECTI DAMT and NCON 2021, 249–252. doi:10.1109/ECTIDAMTNCON51128.2021.9425737.

Huang, G., Liu, Z., Van Der Maaten, L., & Weinberger, K. Q. (2017). Densely connected convolutional networks. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, United States. doi:10.1109/CVPR.2017.243.

Ahmed, S. I., Ibrahim, M., Nadim, M., Rahman, M. M., Shejunti, M. M., Jabid, T., & Ali, M. S. (2023). MangoLeafBD: A comprehensive image dataset to classify diseased and healthy mango leaves. Data in Brief, 47, 108941. doi:10.1016/j.dib.2023.108941.

Full Text: PDF

DOI: 10.28991/ESJ-2024-08-01-03


  • There are currently no refbacks.

Copyright (c) 2024 BAHAR UDDIN MAHMUD, Abdullah Al Mamun, Md Jakir Hossen, Guan Yue Hong