Update README.md
Browse files
README.md
CHANGED
@@ -8,59 +8,59 @@ license: gpl-3.0
|
|
8 |
This project is a web application developed using Flask that allows users to upload images of potato or tomato leaves and receive predictions regarding potential diseases. The application utilizes two deep learning models: one trained to classify potato leaf diseases and another for tomato leaf diseases. Both models were trained using convolutional neural networks (CNNs) and implemented using PyTorch.
|
9 |
|
10 |
## Key Features
|
11 |
-
- Image Upload
|
12 |
-
- Disease Prediction
|
13 |
-
- Dynamic Background
|
14 |
-
- Probability Display
|
15 |
-
|
16 |
## Technologies Used
|
17 |
-
- Python
|
18 |
-
- Flask
|
19 |
-
- PyTorch
|
20 |
-
- HTML/CSS
|
21 |
-
- PIL (Pillow)
|
22 |
-
- OpenCV
|
23 |
-
- Torchvision
|
24 |
-
|
25 |
## Models
|
26 |
-
1. Potato Disease Classification Model
|
27 |
-
- Classes
|
28 |
Potato Early Blight
|
29 |
Potato Late Blight
|
30 |
Potato Healthy
|
31 |
-
- Techniques Used
|
32 |
Convolutional layers for feature extraction.
|
33 |
Batch normalization and max pooling for enhanced training stability and performance.
|
34 |
Dropout layers to prevent overfitting.
|
35 |
|
36 |
-
2. Tomato Disease Classification Model
|
37 |
-
- Classes
|
38 |
Tomato Early Blight
|
39 |
Tomato Late Blight
|
40 |
Tomato Healthy
|
41 |
-
- Techniques Used
|
42 |
Similar architecture to the potato model with appropriate adjustments for tomato disease classification.
|
43 |
Batch normalization, max pooling, and dropout layers are also used here.
|
44 |
|
45 |
## Web Application
|
46 |
-
- Backend
|
47 |
The backend of the application is powered by Flask. It loads the trained models, handles image uploads, processes images, and returns predictions to the frontend.
|
48 |
|
49 |
-
- Model Loading
|
50 |
-
- Prediction Logic
|
51 |
-
- Dynamic Background
|
52 |
|
53 |
-
- Frontend
|
54 |
The frontend is developed using HTML and CSS, with Bootstrap for responsive design.
|
55 |
|
56 |
-
- File Upload Interface
|
57 |
-
- Prediction Display
|
58 |
-
- Dynamic Background
|
59 |
|
60 |
## Outcome
|
61 |
-
- Performance
|
62 |
-
- Potato Model
|
63 |
-
- Tomato Model
|
64 |
-
- Benefits
|
65 |
-
- Disease Detection
|
66 |
-
- User-Friendly Interface
|
|
|
8 |
This project is a web application developed using Flask that allows users to upload images of potato or tomato leaves and receive predictions regarding potential diseases. The application utilizes two deep learning models: one trained to classify potato leaf diseases and another for tomato leaf diseases. Both models were trained using convolutional neural networks (CNNs) and implemented using PyTorch.
|
9 |
|
10 |
## Key Features
|
11 |
+
- **Image Upload:** Users can upload images of potato or tomato leaves.
|
12 |
+
- **Disease Prediction:** The application predicts whether the leaf is healthy or affected by specific diseases.
|
13 |
+
- **Dynamic Background:** The background image of the web page dynamically changes based on whether the user selects potato or tomato.
|
14 |
+
- **Probability Display:** The probability of the predicted class is displayed as a percentage.
|
15 |
-
|
16 |
## Technologies Used
|
17 |
+
- **Python:** Core programming language used for model development and Flask backend.
|
18 |
+
- **Flask:** Web framework for developing the web application.
|
19 |
+
- **PyTorch:** Deep learning framework used to develop and train the models.
|
20 |
+
- **HTML/CSS:** For creating the frontend of the web application.
|
21 |
+
- **PIL (Pillow):** For image processing.
|
22 |
+
- **OpenCV:** For image display and preprocessing.
|
23 |
+
- **Torchvision:** For image transformation utilities.
|
24 |
-
|
25 |
## Models
|
26 |
+
1. **Potato Disease Classification Model**
|
27 |
+
- **Classes:**
|
28 |
Potato Early Blight
|
29 |
Potato Late Blight
|
30 |
Potato Healthy
|
31 |
+
- **Techniques Used:**
|
32 |
Convolutional layers for feature extraction.
|
33 |
Batch normalization and max pooling for enhanced training stability and performance.
|
34 |
Dropout layers to prevent overfitting.
|
35 |
|
36 |
+
2. **Tomato Disease Classification Model**
|
37 |
+
- **Classes:**
|
38 |
Tomato Early Blight
|
39 |
Tomato Late Blight
|
40 |
Tomato Healthy
|
41 |
+
- **Techniques Used:**
|
42 |
Similar architecture to the potato model with appropriate adjustments for tomato disease classification.
|
43 |
Batch normalization, max pooling, and dropout layers are also used here.
|
44 |
|
45 |
## Web Application
|
46 |
+
- **Backend**
|
47 |
The backend of the application is powered by Flask. It loads the trained models, handles image uploads, processes images, and returns predictions to the frontend.
|
48 |
|
49 |
+
- **Model Loading:** Both models (potato and tomato) are loaded at the start of the application to minimize latency during prediction.
|
50 |
+
- **Prediction Logic:** Depending on the selected plant type (potato or tomato), the corresponding model is used to predict the disease class.
|
51 |
+
- **Dynamic Background:** The background image on the frontend changes based on the selected plant type, enhancing user experience.
|
52 |
|
53 |
+
- **Frontend**
|
54 |
The frontend is developed using HTML and CSS, with Bootstrap for responsive design.
|
55 |
|
56 |
+
- **File Upload Interface:** Users can upload an image of a leaf.
|
57 |
+
- **Prediction Display:** After processing, the application displays the predicted disease class and the associated probability.
|
58 |
+
- **Dynamic Background:** The background image changes depending on whether the user is predicting for potato or tomato.
|
59 |
|
60 |
## Outcome
|
61 |
+
- **Performance**
|
62 |
+
- **Potato Model:** Achieved an accuracy of 95% on the validation set, with strong performance in classifying Early Blight, Late Blight, and Healthy leaves.
|
63 |
+
- **Tomato Model:** Achieved an accuracy of 93% on the validation set, effectively distinguishing between Early Blight, Late Blight, and Healthy leaves.
|
64 |
+
- **Benefits**
|
65 |
+
- **Disease Detection:** Helps farmers and agriculturists detect diseases in potato and tomato plants early, potentially preventing crop losses.
|
66 |
+
- **User-Friendly Interface:** The web application provides a simple interface for non-technical users to diagnose plant diseases.
|