athulnambiar
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,49 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
## Potato Blight Disease Prediction Model
|
2 |
+
|
3 |
+
## Overview
|
4 |
+
|
5 |
+
This repository contains a TensorFlow Keras Convolutional Neural Network (CNN) model for predicting Potato Blight Disease. The model is trained to classify images of potato plants as either healthy or affected by blight disease.
|
6 |
+
|
7 |
+
# Dataset
|
8 |
+
|
9 |
+
The dataset used for training and evaluation consists of images of potato plants collected from various sources. Each image is labeled as either "healthy" or "blight" to facilitate supervised learning.
|
10 |
+
|
11 |
+
# Model Architecture
|
12 |
+
|
13 |
+
The CNN model architecture used for this task consists of multiple convolutional and pooling layers followed by fully connected layers. The architecture is designed to effectively capture spatial patterns and features in the input images.
|
14 |
+
|
15 |
+
# Usage
|
16 |
+
|
17 |
+
# Training
|
18 |
+
To train the model, use the provided script train.py. Make sure to specify the appropriate paths for the dataset and any desired hyperparameters.
|
19 |
+
|
20 |
+
bash
|
21 |
+
Copy code
|
22 |
+
python train.py --dataset_path /path/to/dataset --epochs 50 --batch_size 32
|
23 |
+
Evaluation
|
24 |
+
You can evaluate the trained model using the evaluate.py script. This script computes accuracy metrics on a separate test dataset.
|
25 |
+
|
26 |
+
bash
|
27 |
+
Copy code
|
28 |
+
python evaluate.py --model_path /path/to/saved_model --test_dataset_path /path/to/test_dataset
|
29 |
+
Inference
|
30 |
+
To make predictions on new images using the trained model, utilize the predict.py script.
|
31 |
+
|
32 |
+
bash
|
33 |
+
Copy code
|
34 |
+
python predict.py --model_path /path/to/saved_model --image_path /path/to/image
|
35 |
+
Pretrained Model
|
36 |
+
|
37 |
+
A pretrained version of the model is available for download here. You can use this model directly for inference or fine-tuning on your own dataset.
|
38 |
+
|
39 |
+
# Results
|
40 |
+
|
41 |
+
The model achieves an accuracy of X% on the test dataset, demonstrating its effectiveness in predicting potato blight disease.
|
42 |
+
|
43 |
+
# License
|
44 |
+
|
45 |
+
This project is licensed under the MIT License - see the LICENSE file for details.
|
46 |
+
|
47 |
+
# Acknowledgements
|
48 |
+
|
49 |
+
We would like to thank the contributors to the dataset used for training this model.
|