MAU-Net: Full-period Mango Leaf Disease Image Segmentation Algorithm Based on an Improved UNet Network
DOI:
https://doi.org/10.53469/jrse.2025.07(01).18Keywords:
Full-period mango leaf disease, MAU-Net, Leaf Segmentation, Disease SegmentationAbstract
Mango leaf disease segmentation is an essential foundation for accurate disease diagnosis and intelligent grading. The size and shape of mango leaf diseases vary significantly at different times, making it difficult for mainstream semantic segmentation methods to segment disease areas accurately. Therefore, this paper proposes a method called MAU-Net for fine segmentation of mango leaf diseases over the whole period. The MAU-Net is based on the traditional Unet architecture, integrates the Self-Aligning Attention Feature Fusion (SAFF) module and the Multiscale Feature Enhancement (MFE) module, and designs a new loss function DF_Loss. Specifically, the designed SAFF module changes the traditional Unet's skip-connection approach by fusing the global and local two-branch attention mechanisms. It enhances the attention to crucial leaf and disease features at different levels and thus retains richer semantic information about mango leaf diseases. The designed MFE module aims to solve the problem of complex multi-scale disease segmentation in different periods of mango leaves by introducing different scales of cavity convolution to enhance the extraction of disease features at different scales. The designed DF_Loss combines the idea of the similarity measure in Dice Loss and the advantages of the attentional conditioning mechanism in Focal Loss with an additional conditioning factor. It allows the model to focus more on pixels that are difficult to categorize during the learning process, thus improving the segmentation accuracy. MAU-Net achieved 99.21%, 84.33%, 97.1%, and 96.94% of leaf IoU, disease IoU, F1, and mPA metrics on the mango leaf disease dataset. It improved 0.36%, 4.88%, 3.9%, and 1.91% over UNet, and 5.59%, 0.19%, 1.6%, and 2.26% over DeepLabv3+, respectively. Therefore, the present study may provide an accurate method for segmenting mango leaf spots over the whole period and provide a sufficient basis for the accurate analysis of mango leaf diseases.
References
Lebaka, V.R., Wee, Y.-J., Ye, W., Korivi, M.: Nutritional composition and bioactive compounds in three different parts of mango fruit. International Journal of Environmental Research and Public Health 18(2), 741 (2021)
Saravanan, T., Jagadeesan, M., Selvaraj, P., Aravind, M., Raj, G.D., Lokesh, P.: Prediction of mango leaf diseases using convolutional neural network. In: 2023 International Conference on Computer Communication and Informatics (ICCCI), pp. 1–4 (2023). IEEE
Huan Cao, R.F.: Deep learning based classification and identification of mango pests and diseases (in chinese). Computer Technology and Development 33, 115–119 (2023)
Islam, M., Mannan, M., Kabir, M., Islam, A., Olival, K.: Analgesic, antiinflammatory and antimicrobial effects of ethanol extracts of mango leaves. Journal of the Bangladesh Agricultural University 8, 239–244 (2010)
Gulavnai, S., Patil, R.: Deep learning for image based mango leaf disease detection. International Journal of Recent Technology and Engineering (IJRTE) 8(3S3), 54–56 (2019)
Sharma, R.K., Ghandi, P.: Reliability estimation and optimization: A neuro fuzzy based approach. International Journal of Computer Science and Information Security (IJCSIS) 16, 128–132 (2018)
Mu, H., Wang, K., Yang, X., Xu, W., Liu, X., Ritsema, C.J., Geissen, V.: Pesticide usage practices and the exposure risk to pollinators: a case study in the north china plain. Ecotoxicology and Environmental Safety 241, 113713 (2022)
Luo, Y., Sun, J., Shen, J., Wu, X., Wang, L., Zhu, W.: Apple leaf disease recognition and sub-class categorization based on improved multi-scale feature fusion network. IEEE Access 9, 95517–95527 (2021)
Arya, S., Singh, R.: A comparative study of cnn and alexnet for detection of disease in potato and mango leaf. In: 2019 International Conference on Issues and Challenges in Intelligent Computing Techniques (ICICT), vol. 1, pp. 1–6 (2019). IEEE
Merchant, M., Paradkar, V., Khanna, M., Gokhale, S.: Mango leaf deficiency detection using digital image processing and machine learning. In: 2018 3rd International Conference for Convergence in Technology (I2CT), pp. 1–3 (2018). IEEE
Prasetyo, E., Adityo, R.D., Suciati, N., Fatichah, C.: Mango leaf image segmentation on hsv and ycbcr color spaces using otsu thresholding. In: 2017 3rd International Conference on Science and Technology - Computer (ICST), pp. 99–103 (2017). IEEE
Tumang, G.S.: Pests and diseases identification in mango using matlab. In: 2019 5th International Conference on Engineering, Applied Sciences and Technology (ICEAST), pp. 1–4 (2019). IEEE
Pansy, D.L., Murali, M.: Uav hyperspectral remote sensor images for mango plant disease and pest identification using md-fcm and xcs-rbfnn. Environmental Monitoring and Assessment 195(9), 1120 (2023)
Saleem, R., Shah, J.H., Sharif, M., Yasmin, M., Yong, H.-S., Cha, J.: Mango leaf disease recognition and classification using novel segmentation and vein pattern technique. Applied Sciences 11(24), 11901 (2021)
Chouhan, S.S., Singh, U.P., Jain, S.: Web facilitated anthracnose disease segmentation from the leaf of mango tree using radial basis function (rbf) neural network. Wireless Personal Communications 113, 1279–1296 (2020)
Gautam, V., Ranjan, R.K., Dahiya, P., Kumar, A.: Esdnn: A novel ensembled stack deep neural network for mango leaf disease classification and detection. Multimedia Tools and Applications, 1–27 (2023)
Pham, T.N., Van Tran, L., Dao, S.V.T.: Early disease classification of mango leaves using feed-forward neural network and hybrid metaheuristic feature selection. IEEE Access 8, 189960–189973 (2020)
Chouhan, S.S., Singh, U.P., Kaul, A., Jain, S.: A data repository of leaf images: Practice towards plant conservation with plant pathology. In: 2019 4th International Conference on Information Systems and Computer Networks (ISCON), pp. 700–707 (2019). IEEE
Ronneberger, O., Fischer, P., Brox, T.: U-net: Convolutional networks for biomedical image segmentation. In: Medical Image Computing and Computer-Assisted Intervention–MICCAI 2015: 18th International Conference, Munich, Germany, October 5-9, 2015, Proceedings, Part III 18, pp. 234–241 (2015). Springer
Dai, Y., Gieseke, F., Oehmcke, S., Wu, Y., Barnard, K.: Attentional feature fusion. In: Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision, pp. 3560–3569 (2021)
Chen, L.-C., Papandreou, G., Kokkinos, I., Murphy, K., Yuille, A.L.: Deeplab: Semantic image segmentation with deep convolutional nets, atrous convolution, and fully connected crfs. IEEE transactions on pattern analysis and machine intelligence 40(4), 834–848 (2017)
Chen, L.-C., Zhu, Y., Papandreou, G., Schroff, F., Adam, H.: Encoder-decoder with atrous separable convolution for semantic image segmentation. In: Proceedings of the European Conference on Computer Vision (ECCV), pp. 801–818 (2018)
Hu, J., Shen, L., Sun, G.: Squeeze-and-excitation networks. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 7132–7141 (2018)
Long, J., Shelhamer, E., Darrell, T.: Fully convolutional networks for semantic segmentation. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 3431–3440 (2015)
Cao, H., Wang, Y., Chen, J., Jiang, D., Zhang, X., Tian, Q., Wang, M.: Swinunet: Unet-like pure transformer for medical image segmentation. In: European Conference on Computer Vision, pp. 205–218 (2022). Springer
Zhao, H., Shi, J., Qi, X., Wang, X., Jia, J.: Pyramid scene parsing network. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2881–2890 (2017)
Paszke, A., Chaurasia, A., Kim, S., Culurciello, E.: Enet: A deep neural network
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2025 Wenkang Tang, Bibo Lu, Peipei Zhou, Jie Yang, Aiqing Song
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
