Sudheer17 commited on
Commit
df5164e
Β·
verified Β·
1 Parent(s): 23890f6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +223 -0
README.md CHANGED
@@ -0,0 +1,223 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ title: AI Pet Classifier
3
+ emoji: 🐢
4
+ colorFrom: blue
5
+ colorTo: indigo
6
+ sdk: streamlit
7
+ sdk_version: "1.46.1"
8
+ app_file: app.py
9
+ pinned: false
10
+ license: mit
11
+ ---
12
+
13
+ # 🐢 AI Pet Classifier using CNN
14
+
15
+ A deep learning application that classifies images of cats and dogs using a Convolutional Neural Network (CNN) built with TensorFlow and Keras.
16
+
17
+ ...
18
+
19
+ # 🐢🐱 AI Pet Classifier using Convolutional Neural Networks (CNN)
20
+
21
+ ## Overview
22
+
23
+ AI Pet Classifier is a deep learning project that classifies images as either **Cat** or **Dog** using a Convolutional Neural Network (CNN) built with TensorFlow and Keras. The model is trained on thousands of labeled pet images and predicts the class of unseen images with high accuracy.
24
+
25
+ ---
26
+
27
+ ## Features
28
+
29
+ - Binary Image Classification
30
+ - TensorFlow & Keras Implementation
31
+ - Data Augmentation
32
+ - Batch Normalization
33
+ - Image Preprocessing
34
+ - Model Saving & Loading
35
+ - Single Image Prediction
36
+ - Beginner-Friendly Notebook
37
+ - Google Colab Compatible
38
+
39
+ ---
40
+
41
+ ## Project Pipeline
42
+
43
+ ```
44
+ Dataset
45
+ β”‚
46
+ β–Ό
47
+ Image Preprocessing
48
+ β”‚
49
+ β–Ό
50
+ Data Augmentation
51
+ β”‚
52
+ β–Ό
53
+ CNN Model
54
+ β”‚
55
+ β–Ό
56
+ Training
57
+ β”‚
58
+ β–Ό
59
+ Evaluation
60
+ β”‚
61
+ β–Ό
62
+ Prediction
63
+ ```
64
+
65
+ ---
66
+
67
+ ## CNN Architecture
68
+
69
+ ```
70
+ Input Layer (64Γ—64Γ—3)
71
+
72
+ ↓
73
+
74
+ Conv2D (32 Filters)
75
+ ↓
76
+
77
+ Batch Normalization
78
+ ↓
79
+
80
+ MaxPooling
81
+
82
+ ↓
83
+
84
+ Conv2D (64 Filters)
85
+ ↓
86
+
87
+ MaxPooling
88
+
89
+ ↓
90
+
91
+ Conv2D (128 Filters)
92
+ ↓
93
+
94
+ MaxPooling
95
+
96
+ ↓
97
+
98
+ Flatten
99
+
100
+ ↓
101
+
102
+ Dense (128)
103
+
104
+ ↓
105
+
106
+ Dense (1, Sigmoid)
107
+
108
+ ↓
109
+
110
+ Prediction
111
+ ```
112
+
113
+ ---
114
+
115
+ ## Technologies Used
116
+
117
+ - Python
118
+ - TensorFlow
119
+ - Keras
120
+ - NumPy
121
+ - Matplotlib
122
+ - Pillow
123
+ - Google Colab
124
+
125
+ ---
126
+
127
+ ## Dataset Structure
128
+
129
+ ```
130
+ Data/
131
+
132
+ β”œβ”€β”€ training_set/
133
+ β”‚ β”œβ”€β”€ cats/
134
+ β”‚ └── dogs/
135
+ β”‚
136
+ └── test_set/
137
+ β”œβ”€β”€ cats/
138
+ └── dogs/
139
+ ```
140
+
141
+ ---
142
+
143
+ ## Hyperparameters
144
+
145
+ | Parameter | Value |
146
+ |------------|-------|
147
+ | Image Size | 64 Γ— 64 |
148
+ | Batch Size | 32 |
149
+ | Epochs | 25 |
150
+ | Optimizer | Adam |
151
+ | Loss Function | Binary Crossentropy |
152
+ | Activation | ReLU |
153
+ | Output Activation | Sigmoid |
154
+
155
+ ---
156
+
157
+ ## Training
158
+
159
+ The model uses image augmentation to improve generalization by applying:
160
+
161
+ - Rescaling
162
+ - Random Zoom
163
+ - Shear Transformation
164
+ - Horizontal Flip
165
+
166
+ ---
167
+
168
+ ## Prediction
169
+
170
+ The trained model predicts whether the uploaded image belongs to:
171
+
172
+ - 🐱 Cat
173
+ - 🐢 Dog
174
+
175
+ along with the prediction confidence.
176
+
177
+ ---
178
+
179
+ ## Future Improvements
180
+
181
+ - Early Stopping
182
+ - Model Checkpoint
183
+ - Transfer Learning (MobileNetV2 / EfficientNet)
184
+ - Confusion Matrix
185
+ - Classification Report
186
+ - Accuracy & Loss Curves
187
+ - Grad-CAM Visualization
188
+ - Streamlit & Hugging Face Deployment
189
+
190
+ ---
191
+
192
+ ## Repository Structure
193
+
194
+ ```
195
+ β”œβ”€β”€ CNN_Model.ipynb
196
+ β”œβ”€β”€ cnn_model.keras
197
+ β”œβ”€β”€ Data.zip
198
+ β”œβ”€β”€ requirements.txt
199
+ β”œβ”€β”€ README.md
200
+ └── LICENSE
201
+ ```
202
+
203
+ ---
204
+
205
+ ## Author
206
+
207
+ **Sudheer Muthyala**
208
+
209
+ B.Tech (ECE)
210
+
211
+ Aspiring AI & Data Science Engineer
212
+
213
+ GitHub: https://github.com/M-Sudheer18
214
+
215
+ ---
216
+
217
+ ## License
218
+
219
+ This project is licensed under the MIT License.
220
+
221
+ ---
222
+
223
+ ⭐ If you found this project helpful, consider giving it a star.