Update README.md
Browse files
README.md
CHANGED
@@ -14,11 +14,35 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
|
|
14 |
|
15 |
# Butterfly Classification using CNN
|
16 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
17 |
## 1. Introduction
|
18 |
This project focuses on classifying butterfly images using a Convolutional Neural Network (CNN). The dataset includes images of butterflies and moths, with the goal of training a model to accurately predict the species of a given butterfly image.
|
19 |
|
20 |
## 2. Model Architecture
|
21 |
The CNN model used for this project is based on the ResNet50V2 architecture. The model is trained on a dataset consisting of training, validation, and test sets. The key components of the model architecture include convolutional layers, dropout for regularization, global average pooling, and fully connected layers. The model is compiled using the Adam optimizer and sparse categorical crossentropy loss.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
22 |
|
23 |
## 3. How-to Guide
|
24 |
### 3.1. Data Preparation
|
@@ -45,8 +69,22 @@ The CNN model used for this project is based on the ResNet50V2 architecture. The
|
|
45 |
- Gradio is utilized for deploying the trained model.
|
46 |
- Users can input an image, and the model will predict the butterfly species.
|
47 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
48 |
## 4. License
|
49 |
This project is licensed under the [MIT License](LICENSE).
|
50 |
|
51 |
## 5. Contributors
|
52 |
-
- Kamelia Zaman Moon
|
|
|
14 |
|
15 |
# Butterfly Classification using CNN
|
16 |
|
17 |
+
- **Author:** Kamelia Zaman Moon
|
18 |
+
- **Project link:** https://huggingface.co/spaces/KameliaZaman/Butterfly-Classification-using-CNN
|
19 |
+
- **Language(s):** Python
|
20 |
+
- **License:** MIT
|
21 |
+
- **Contact:** [email protected]
|
22 |
+
|
23 |
+
## Table of Contents
|
24 |
+
- [Introduction](#introduction)
|
25 |
+
- [Model Architecture](#model-architecture)
|
26 |
+
- [How-to Guide](#how-to-guide)
|
27 |
+
- [License](#license)
|
28 |
+
- [Contributors](#contributors)
|
29 |
+
|
30 |
## 1. Introduction
|
31 |
This project focuses on classifying butterfly images using a Convolutional Neural Network (CNN). The dataset includes images of butterflies and moths, with the goal of training a model to accurately predict the species of a given butterfly image.
|
32 |
|
33 |
## 2. Model Architecture
|
34 |
The CNN model used for this project is based on the ResNet50V2 architecture. The model is trained on a dataset consisting of training, validation, and test sets. The key components of the model architecture include convolutional layers, dropout for regularization, global average pooling, and fully connected layers. The model is compiled using the Adam optimizer and sparse categorical crossentropy loss.
|
35 |
+
```
|
36 |
+
ββ model_checkpoint_manual_effnet.h5 - generated model.
|
37 |
+
|
38 |
+
ββ Butterfly classification with CNN.ipynb - preprocesses input, trains, saves and evaluates the model.
|
39 |
+
|
40 |
+
ββ app.py - this module starts the app interface.
|
41 |
+
|
42 |
+
ββ README.md - readme file of this project.
|
43 |
+
|
44 |
+
ββ requirements.txt - list of required packages.
|
45 |
+
```
|
46 |
|
47 |
## 3. How-to Guide
|
48 |
### 3.1. Data Preparation
|
|
|
69 |
- Gradio is utilized for deploying the trained model.
|
70 |
- Users can input an image, and the model will predict the butterfly species.
|
71 |
|
72 |
+
```bash
|
73 |
+
# clone project
|
74 |
+
git clone https://huggingface.co/spaces/KameliaZaman/Butterfly-Classification-using-CNN/tree/main
|
75 |
+
|
76 |
+
# go inside the project directory
|
77 |
+
cd Butterfly-Classification-using-CNN
|
78 |
+
|
79 |
+
# install the required packages
|
80 |
+
pip install -r requirements.txt
|
81 |
+
|
82 |
+
# run the gradio app
|
83 |
+
python app.py
|
84 |
+
```
|
85 |
+
|
86 |
## 4. License
|
87 |
This project is licensed under the [MIT License](LICENSE).
|
88 |
|
89 |
## 5. Contributors
|
90 |
+
- Kamelia Zaman Moon - [email protected]
|