Development of a Pomegranate Fruit Disease Detection and Classification Model using Deep Learning

Pomegranates are a nutritious and versatile fruit with significant health benefits, culinary uses and economic importance (Singh et al., 2023). Cultivating pomegranates requires specific climate and soil conditions, along with effective pest and water management. Despite challenges, pomegranates remain a valuable crop for farmers and a popular fruit for consumers worldwide (Pakruddin et al., 2024; Gawade et al., 2018). The import and export of pomegranates play a vital role in India’s agricultural economy, providing income and employment while contributing to foreign exchange earnings. However, challenges such as quality control, pest management, infrastructure, market access and climate change must be addressed to capitalize on this potential fully. Effective government policies, industry initiatives and investment in infrastructure and technology are crucial to overcoming these challenges and boosting the pomegranate trade Wakhare et al., (2023). Pomegranate plants can be affected by a diversity of diseases, which can meaningfully impact yield and fruit quality. These diseases can be classified into fungal, bacterial, viral and other categories.
       
Fungal disease and its symptoms Alternaria Fruit Rot (Alternaria alternate): Dark, sunken lesions on fruits, the internal tissue may become discolored and rot Aloi and Francesco et al., (2021). Cercospora Fruit Spot (Cercospora punicae): Small, dark spots on leaves and fruit, spots may coalesce to form larger lesions (Gk Ravichandra et al., 2023). Aspergillus Fruit Rot (Aspergillus niger): lack mold on the surface of the fruit, affected fruit may shrivel and become hard. Botrytis Fruit Rot (Botrytis cinerea): Gray mold on flowers and fruits, can lead to fruit drop and reduced yield. Fusarium Wilt (Fusarium oxysporum): Yellowing and wilting of leaves, vascular discoloration and stunted growth (Nirgude et al., 2024; AlZubi et al., 2024). Bacterial Diseases and its Signs Bacterial Blight (Xanthomonas axonopodis pv. Punicae): Fruit cracking can result from water-soaked lesions on leaves, stems and fruits that turn black. (Sharath et al., 2019). Viral Diseases and its Symptoms Pomegranate Mosaic Virus Mosaic patterns on leaves, leaf distortion and reduced fruit size and quality. Other Diseases and their Symptoms Heart Rot (Phomopsis sp., Colletotrichum gloeosporioides): Internal fruit decay, often without external symptoms, affected fruits may have a hollow or rotten core. Leaf Spot (Myrothecium roridum): Circular, brown spots with a yellow halo on leaves, severe infections can lead to defoliation.
       
Farmers may efficiently control pomegranate diseases and ensure healthier plants and higher-quality fruit output by putting these methods into practice. Improve air circulation, ensure proper plant spacing, maintain field hygiene and avoid overhead irrigation. Use fungicides and bactericides as necessary, following recommended guidelines for application. Employ beneficial microorganisms that can suppress disease-causing pathogens. Plant disease-resistant varieties whenever possible to reduce the impact of diseases. To stop infections from spreading, remove and destroy any contaminated plant parts. Fruits should be stored cool and dry to avoid postharvest illnesses.
       
After employing machine learning and deep learning techniques to perform a survey on illness detection and classification for pomegranate fruit, we have concluded that additional study is required to address developing diseases. The proposed model addresses the deficiencies of prior approaches in deep learning-based pomegranate fruit disease detection and classification. The existing model for pomegranate fruit disease was proposed by the authors, but it had several drawbacks, including low accuracy and high processing times. To address these issues, we proposed a PomeNetV2 model that predicts pomegranate diseases much more quickly and accurately than the current system. This paper’s major goal is to create a deep learning-based automated system that can reliably identify and categorize pomegranate fruit disorders. The goal of the study is to evaluate and confirm the effectiveness of the system, which will aid in the creation of useful instruments for identifying fruit diseases and lessen the detrimental effects of plant diseases on crop quality and yield.
       
The researchers (Naseer et al., 2024) investigated identifying pomegranate growth stages early on by employing an effective method. According to the results, the random forest model outperformed cutting-edge techniques, achieving a high accuracy of 98%. Farmers that use this strategy can reduce risks and increase crop productivity.
       
Nirgude et al., (2022) suggest a framework for deploying agricultural drones and sensors to collect data in real-time on weather, soil and water characteristics. Random Forest (RF), out of all the machine learning models that were assessed, had the best accuracy (96.53%). By using this method, the pomegranate industry can better diagnose and classify prevalent diseases, leading to improved fruit quality and growth.
       
A CNN-LSTM deep learning model was created by the authors Vasumathi et al., (2021) to divide a dataset of 6519 fruits into categories for normal and pathological conditions. Each dataset entry contains nine features within an Excel file. CNNs perform deep feature extraction and LSTMs use these features for categorization. With an accuracy of 98.17%, sensitivity of 97.77%, specificity of 98.65% and an F1-score of 98.39%, the suggested system performed well. Previous detection methods had lower sensitivity and accuracy due to unresolved faults. Deep learning offers advanced techniques to enhance analytical and predictive accuracy in early illness identification.
       
The purpose of the study by Aloi et al., (2021) was to define the mycotoxigenic characteristics of Alternaria species that cause heart rot in pomegranates in southern Italy. The CNN-based ResNet50 architecture proposed by the authors Nirgude et al., (2021) effectively identifies and classifies five distinct illness types with ambiguous symptoms and complex histories. The enhanced ResNet50 model attained a test accuracy of 97.92%, outperforming both Inception-V3 (78.75%) and ResNet18 (87.5%) at a learning rate of 0.001. The multiclass cross-entropy loss function was used to determine the error rate.
       
Patil et al., (2021) explain that their study examines feature extraction, segmentation, preprocessing and classification strategies for pomegranate plant disease detection. It observes the shortcomings of current approaches for detecting fruit diseases and compares them. Vasumathi et al., (2021) state that a hybrid CNN-LSTM approach to identify and categorize four different pomegranate fruit illnesses. Discriminant analysis and principal component analysis are used to extract and process features such as color, texture and shape. By adjusting the cost function and optimizing the weight, the dragonfly algorithm raises the initial 92% accuracy of the CNN-LSTM classifier to 97.1%. With a graphical user interface, Mangena et al., (2021) suggested MATLAB-based framework successfully classifies damaged and healthy leaves with an accuracy of 98.39%. The findings of experiments also show that it achieves 98.07% accuracy in diagnosing illnesses on pomegranate leaves.
       
The primary goal of this paper is to develop an effective PomeNetV2 model for accurately detecting pomegranate fruit diseases using advanced deep learning techniques, a specially designed dataset, activation functions and optimization algorithms. The study will assess and validate the model’s performance in identifying and classifying these diseases. By accomplishing this, the research seeks to contribute significantly to the creation of efficient tools for detecting fruit diseases, ultimately helping to mitigate their adverse effects on crop yield and quality.

All the experiments were conducted at the research center named Rashtreeya Vidyalaya College of Engineering Bangalore in the academic year 2023-2024.
 
Dataset used
 
In this research, we have leveraged a newly proposed dataset B Pakruddin et al., (2024) focusing on the afflictions affecting pomegranate fruits. This dataset is a culmination of diverse samples sourced from various regions such as Ballari, Bengaluru and Bagalakote, among others, spanning from July to October 2023. Comprising a total of 5099 meticulously curated images, this dataset encompasses a comprehensive array of pomegranate fruit conditions, exactly categorized into five distinct types: Healthy (1450 images), Bacterial blight (966 images), Anthracnose (1166 images), Chavan et al., (2017). Cercospora (631 images) and Alternaria (886 images). Each image is accurately labeled and organized into corresponding folders, ensuring seamless access and utilization. Samples of pomegranate fruit disease images are exposed in Fig 1.
 

 
Image pre-processing and labelling
 
Image resizing
 
Image resizing in the pre-processing step is essential for ensuring uniform input size, reducing computational load, ensuring model compatibility, maintaining consistency and improving overall model performance. These benefits are critical for effective and efficient model training and testing. Each image size in the dataset contains 3120 * 3120, during pre-processing each image size is considered as 220 * 220.
 
Pixel normalization
 
We should resize the photographs to the proper size before feeding them into the network. Additionally, to normalize the pixel values and improve the model¢s performance, we should divide them by 256 and scale them to the range of 0 to 1. Our model can be pre-processed by adding a layer and this normalization should be implemented during both training and inference (Bhange et al., 2015).
 
Noise removal
 
Noise removal is an essential pre-processing step to improve image quality, reduce irrelevant information and enhance feature extraction. Techniques such as Gaussian blurring, median filtering, bilateral filtering and non-local means denoising are commonly used to effectively remove noise while preserving important image details.
 
Augmentation
       
Data augmentation is essential for increasing dataset size, improving model robustness, preventing overfitting and balancing classes.
 
Mathematical formulation of PomeNetV2 CNN model
 
Six convolutional layers, six max-pooling layers Thorat et al., (2023), one flattened layer and two dense layers make up our PomeNetV2 deep learning model. The mathematical notation for this model is organized as follows:
1. Convolutional and max-pooling layers.
                    
2. Flatten layer  
3. Dense layers
                           
4. Output layer
                                                                   
Where
X: Input tensor.
Wi, bi: Weight matrix and bias vector for the j-^th convolutional layer.
Wi,^d bi^d: Weight matrix and bias vector for the j-^th dense layer.
f: Activation function (e.g., ReLU for convolutional layers, softmax for output layer).
P: Max-pooling operation, *: Convolution operation Akhilesh et al. (2020).

Performance metrics
 
The effectiveness of deep learning models can be measured using metrics such as Accuracy, Precision, Recall, F1 score and Support. These metrics deliver an excellent means of comprehending several facets of the model’s performance (Vasumathi et al., 2021; Khatawkar et al., 2023).
 
Support: Support represents the number of occurrences of each class in actual data. It gives an indication of the frequency of each class in the dataset.
       
Where the confusion matrix is represented by true positives (TP), true negatives (TN), false positives (FP) and false negatives (FN) values.
 
Experimental setup for the proposed model
 
The research was conducted utilizing Google Colab, utilizing a system with 12.7 GB of RAM and 107.7GB of disk space, employing the Python programming language. This was executed on a Windows 10, 64-bit i3 computer equipped with 8GB of RAM. The performance of the proposed PomeNetV2 Convolutional Neural Network (CNN) model was evaluated against various existing methods across different types of pomegranate fruit diseases. Table 1 Summarizes some common hyperparameters used for the PomeNetV2 model.
 

       
In the realm of deep learning, data augmentation is a commonly employed tactic aimed at enhancing model performance and generalization, particularly in scenarios where datasets are limited. The fundamental aim of data augmentation is to introduce diverse transformations to the existing data, thereby expanding the size of the training dataset artificially. This process enhances the model’s capability to generalize effectively to unseen data and facilitates the learning of invariant features. Basic data augmentation techniques such as rotation, flipping, zooming and cropping, among others, find widespread application in various deep learning contexts. Table 2 showcases a selection of methods utilized for both pre-trained models and the proposed model.
 

       
In this study, we have also used three techniques for data augmentation Cache, Shuffle and Prefetch. Cache involves storing the processed dataset in memory (or on disk) after it has been loaded and possibly pre-processed for the first time. This reduces the overhead of repeated data loading and pre-processing during each epoch. Shuffling (buffer_size = 1000) the dataset ensures that the order of the data is randomized. This is essential to boost generalization and stop the model from picking up any erroneous patterns from the data's order. Prefetching (buffer_size = tf.data.AUTOTUNE) overlaps the data pre-processing and model training by preparing the next batch of data while the current batch is being processed. This is done asynchronously to ensure that the data pipeline can keep up with the training process. Fig 2 describes dataset is separated into 3 subsets, namely training, validation and testing with different percentages.
 

 
Proposed model architecture
 
The author developed PomeNetV2, a lightweight 15-layer CNN model designed to classify healthy images and four disease categories within the proposed dataset. This model, built without using transfer learning, includes 6 convolutional layers with trainable weights and biases, 6 max-pooling layers, 1 flattened layer and 2 fully-connected dense layers. To combat overfitting, the model features a reduced number of layers and fewer filters per layer. Each convolutional layer employs a 3×3 kernel size and each max-pooling layer uses a 2×2 pool size, with 64 filters in each layer, a stride of size 1 and valid padding. Feature extraction occurs in the first 12 layers, while the final 3 layers are used for classification. The model is trained from scratch without any transfer learning. Fig 3 describes the visualization of the PomeNetV2 CNN framework, while the input and output dimensions of each layer in the suggested PomeNetV2 CNN structure are shown in Fig 4.
 

 

Table 3 displays the performance outcomes of numerous activation functions and loss functions when paired with the recommended 15-layer PomeNetV2 CNN architecture. Sparse categorical cross entropy outperformed the other loss functions that were investigated in terms of performance. Conversely, binary cross-entropy and categorical cross-entropy loss functions are incompatible due to multiclass classification. This discrepancy can be attributed to binary cross entropy¢s suitability for binary class data, making it less effective for categorizing multiclass pomegranate disease grades. Additionally, three activation functions were tested in this investigation, Softmax and Sigmoid activation functions in conjunction with the sparse categorical cross-entropy loss function achieving accuracy of 99.02% and 99.00% respectively. This setup performed better than every other combination that was looked at. 19.34% accuracy was obtained using the Tanh activation function and the sparse categorical cross-entropy loss function.
 

 
The performance metrics of the PomeNetV2 framework-namely training accuracy, training loss, validation accuracy and validation loss-are meticulously detailed across different epoch counts in Table 4. These metrics provide insights into the model’s learning process and its ability to generalize from the training data. The dataset used for developing the framework is strategically divided into three subsets: training data, testing data and validation data. This division is visually represented in Fig 5, which also indicates that the images for each subset were selected randomly to ensure unbiased training and evaluation. Fig 5 further illustrates the accuracy achieved for each data split. Among various configurations tested, the 80%-10%-10% split-comprising 80% of the data for training, 10% for testing and 10% for validation-proved to be the most effective. This particular split reached a peak accuracy of 99.02% at 75 epochs, demonstrating the framework’s exceptional ability to accurately classify pomegranate diseases and healthy fruit under this data distribution. This high level of performance highlights the robustness of the PomeNetV2 framework and its potential for real-world application in agricultural settings, where precise disease detection is crucial for crop management and yield optimization.
 



 
 The summary indicates that the designed model, PomeNetV2, achieves an overall accuracy of approximately 99.02%. Fig 6 Specifically, for Alternaria, the framework achieves an accuracy of about 99.99%. For Anthracnose, the accuracy is approximately 99.75%. The framework achieves 100% accuracy for Bacterial Blight, Cercospora and Health. The new work must be compared to previously published studies to determine which model can produce greater accuracy. The outcomes are displayed in Table 5. Table 6 displays the additional performance parameters, which include recall, accuracy, precision, F1-score and support. Fig 7 and 8 describe loss and accuracy for training and testing. Fig 9 Shows accuracy achieved 99.02% for 75 epochs.
 





 

The article introduces the PomeNetV2 framework, a sophisticated system designed for the automatic identification and classification of pomegranate diseases through the analysis of fruit images. This innovative framework integrates several advanced techniques, including image pre-processing, augmentation, feature extraction, max pooling and image classification, to accurately diagnose pomegranate conditions. PomeNetV2 specifically targets four types of pomegranate diseases: Alternaria, Anthracnose, Bacterial Blight and Cercospora fruit spot, in addition to identifying healthy fruit. The framework demonstrates impressive performance, achieving an accuracy of approximately 99.99% for Alternaria, 99.75% for Anthracnose and 100% accuracy for both Bacterial Blight nd Cercospora fruit spot, as well as healthy pomegranates. Overall, PomeNetV2 maintains a remarkable accuracy rate of about 99.02% across all categories. Looking ahead, we have planned to create an Android application to further support farmers. This app will enable efficient detection and classification of pomegranate diseases in real-time and provide actionable remedy suggestions, enhancing the overall management and health of pomegranate crops.

Public, commercial, or nonprofit funding agencies did not provide a particular grant for this study.
 
Data availability
 
Pomegranate Fruit Diseases Dataset for Deep Learning Models (Original data) (Mendeley Data).
https://data.mendeley.com/datasets/b6s2rkpmvh/1
 
Credit author statement
 
Pakruddin B: Conceptualization, Formal analysis, Methodology, Investigation and Writing-original draft, Writing -review and editing; Dr. Hemavathy R: Supervision, Validation, Writing-review and editing.

The authors state that none of their known financial conflicts or interpersonal connections might have had an impact on the work presented in this paper.