|
| 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 | + |
| 26 | + |
| 27 | +## 📈 Performance of the Models based on the Accuracy Scores |
| 28 | + |
| 29 | +**CNN Performance** |
| 30 | + |
| 31 | + |
| 32 | + |
| 33 | +**EfficientNet Performance** |
| 34 | + |
| 35 | + |
| 36 | + |
| 37 | +**ResNet50 Performance** |
| 38 | + |
| 39 | + |
| 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