athulnambiar commited on
Commit
c8e4cbb
·
verified ·
1 Parent(s): f80cb7c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +49 -3
README.md CHANGED
@@ -1,3 +1,49 @@
1
- ---
2
- license: mit
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.