Skip to content

Commit 294b8bb

Browse files
authored
Merge pull request #240 from keshripritesh/P2
Pull Request : Automated Plant Seedlings Classification with CNN Arch…
2 parents 5280a40 + 6955754 commit 294b8bb

8 files changed

Lines changed: 1285 additions & 0 deletions

File tree

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
The dataset which is used in this project, is collected from Kaggle. Here is the link of the dataset : https://www.kaggle.com/vbookshelf/v2-plant-seedlings-dataset
79.4 KB
Loading
37.4 KB
Loading
228 KB
Loading
34.7 KB
Loading

Plant Seedlings Classification/Model/Plant_seedlings_classification.ipynb

Lines changed: 1205 additions & 0 deletions
Large diffs are not rendered by default.
Lines changed: 63 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,63 @@
1+
# Plant Seedlings Classification
2+
Agriculture is very important to human continued existence and remains a key driver of many economies worldwide, especially in underdeveloped and developing economies. There is an increasing demand for food and cash crops, due to the increasing in world population and the challenges enforced by climate modifications, there is an urgent need to increase plant production while reducing costs. Preceding instrument vision methods established for selective weeding have confronted with major challenges for trustworthy and precise weed recognition. In this project, plant seedlings classification approach is presented with a dataset that contains approximately 5,000 images with 960 unique plants that belong to 12 species at a few developing phases. Convolutional Neural Network (CNN) algorithms, a deep learning technique extensively applied to image recognition was used, for this task. The results found that CNN-driven seedling classification applications when used in farming automation have the latent to enhance crop harvest and improve output and productivity when designed properly. The trained model achieved an accuracy of 99.48% on a held-out test set, demonstrating the feasibility of this approach.
3+
4+
![](Images/plant1.jpg)
5+
6+
## Goal
7+
The goal of this project is to build the classification model. The architectures that I have used are, ResNet, AlexNet, Vgg, Inception, MobileNet, SqueezeNet, DenseNet, to deploy the classification model.
8+
9+
## Dataset
10+
The dataset which is used in this project, is collected from Kaggle. Here is the link of the dataset : https://www.kaggle.com/vbookshelf/v2-plant-seedlings-dataset
11+
12+
## What Have I done
13+
1. Loading and importing all the libraries, check [`requirements.txt`](requirements.txt).
14+
2. Importing the dataset in the Jupyter Notebook.
15+
3. Then I prepared the Classification model using the Neural Networks.
16+
4. These are following steps -
17+
- Classification Algorithms using Neural Networks
18+
- Setting the model architecture
19+
- Evaluating the deployment of the architectures
20+
- Getting prediction on validation set
21+
- Creating Confusion Matrix
22+
- Plotting the Loss and Accuracy on the Training and Validation set
23+
5. Conclusion
24+
25+
## Libraries used
26+
|Numpy|Pandas|Matplotlib|Tensorflow|Keras|copy|date|time|shutils|torch|
27+
|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|:---:|
28+
29+
## Model Accuracy Visualization
30+
31+
![](Images/plant2.png)
32+
33+
## Model Comparison
34+
I have deployed seven Convolution Neural Network architectures for this Plant Seedlings Classification project. The model is successfully deployed and the accuracy of the model is checked using the accuracy score. CNN is one of the finest neural networks and the architectures are grading up the model to the higher extent.
35+
36+
After evaluating the architectures, the accuracy score of the architectures are shown below -
37+
38+
|Name of the Architecture|Best Accuracy Score|
39+
|:---:|:---:|
40+
|ResNet|0.69|
41+
|AlexNet|0.69|
42+
|MobileNet|0.67|
43+
|vgg|0.68|
44+
|DenseNet|0.67|
45+
|SqueezeNet|0.67|
46+
|Inception|0.68|
47+
48+
## Conclusion
49+
**Comparing all those scores scored by the deep learning algorithms, it is clear that ResNet and AlexNet architecture is having the upper hand in case of this dataset, than rest of the architectures of CNN.**
50+
51+
Best Fitted Models ranking -
52+
1. ResNet
53+
2. AlexNet
54+
3. Inception
55+
4. VGG
56+
5. MobileNet
57+
6. DenseNet
58+
7. SqueezeNet
59+
60+
Hooray!! The models are deployed successfully!
61+
62+
********************************************************************
63+
Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
The required libraries to deploy this project
2+
3+
Tensorflow==2.5.0
4+
OpenCV==4.5.2
5+
Keras==2.4.0
6+
Numpy==1.19.2
7+
Pandas==1.2.4
8+
Matplotlib==3.4.2
9+
Sklearn
10+
Seaborn
11+
torch==10.0
12+
copy
13+
time
14+
date
15+
shutil
16+
itertools

0 commit comments

Comments
 (0)