Skip to content

Commit 1fc8da3

Browse files
authored
Merge pull request #870 from abckhush/main
Added Butterfly Image Classification
2 parents bca1cd7 + a16b07f commit 1fc8da3

10 files changed

+2766
-0
lines changed
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
# **Butterfly Image Classification**
2+
3+
The dataset features 75 different classes of Butterflies. The dataset contains about 1000+ labelled images including the validation images. Each image belongs to only one butterfly category.
4+
5+
### Dataset Link : https://www.kaggle.com/datasets/phucthaiv02/butterfly-image-classification/data
Loading
Loading
Loading
Loading
Loading

Butterfly Image Classification/Models/butterfly-classification-cnn.ipynb

Lines changed: 1253 additions & 0 deletions
Large diffs are not rendered by default.

Butterfly Image Classification/Models/butterfly-classification-efficientnet.ipynb

Lines changed: 1451 additions & 0 deletions
Large diffs are not rendered by default.

Butterfly Image Classification/Models/butterfly-classification-resnet-50.ipynb

Lines changed: 1 addition & 0 deletions
Large diffs are not rendered by default.
Lines changed: 56 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,56 @@
1+
# **Butterfly Image Classification**
2+
3+
## 🎯 Goal
4+
The primary goal of this project is to build and compare various deep learning models to accurately classify butterfly images into their respective species.
5+
6+
## 🧵 Dataset : https://www.kaggle.com/datasets/phucthaiv02/butterfly-image-classification
7+
8+
## 🧾 Description
9+
This dataset consists of over 1000 labeled images of butterflies, including validation images. Each image belongs to only one butterfly category. The challenge is to develop models that can accurately classify these images into the correct species.
10+
11+
## 🚀 Models Implemented
12+
1. Convolutional Neural Network (CNN)
13+
2. EfficientNet
14+
3. ResNet50
15+
16+
## 📚 Libraries Needed
17+
- TensorFlow: For building and training deep learning models.
18+
- Keras: For simplifying the creation and training of neural networks.
19+
- NumPy: For numerical computations and array operations.
20+
- Pandas: For data manipulation and analysis.
21+
- Matplotlib: For plotting and visualizing data.
22+
23+
## 📊 Exploratory Data Analysis Results
24+
The dataset features 75 different classes of Butterflies. The dataset contains about 1000+ labelled images including the validation images. Each image belongs to only one butterfly category.
25+
![Distribution of Butterfly Classes](https://github.com/user-attachments/assets/b274368f-aa5e-4722-85c1-943339d36373)
26+
27+
## 📈 Performance of the Models based on the Accuracy Scores
28+
29+
**CNN Performance**
30+
31+
![CNN Accuracy Plot](https://github.com/user-attachments/assets/e994354e-6693-4e38-87e6-948de0d1c524)
32+
33+
**EfficientNet Performance**
34+
35+
![EfficientNet Accuracy Plot](https://github.com/user-attachments/assets/0f0027ac-47d8-43f6-b4ba-6d9be0e8e876)
36+
37+
**ResNet50 Performance**
38+
39+
![ResNet Accuracy Plot](https://github.com/user-attachments/assets/60d54f34-bf63-4096-9bc4-01994927715e)
40+
41+
## 📢 Conclusion
42+
The following models were implemented and evaluated based on their accuracy scores:
43+
44+
### Accuracy Results
45+
| Model | Accuracy |
46+
|-------|----------|
47+
| CNN | 87.36% |
48+
| ResNet50 | 80.82% |
49+
| EfficientNet | 91.85% |
50+
51+
## Best Fitted Model
52+
EfficientNet achieved the highest accuracy of 91.85%, making it the best-performing model for this butterfly image classification task.
53+
54+
## ✒️ Contributor
55+
- Name : Khushi Kalra
56+
- LinkedIn: https://www.linkedin.com/in/kalrakhushi/

0 commit comments

Comments
 (0)