Vertika-1312's picture
Updated README.md
5af446a verified
|
Raw
History Blame Contribute Delete
1.58 kB
---
license: mit
library_name: keras
tags:
- tensorflow
- keras
- cnn
- computer-vision
- image-classification
- dogs-vs-cats
- binary-classification
---
# 🐢🐱 Dog vs Cat Image Classifier
## πŸ“Œ Overview
This repository contains a Convolutional Neural Network (CNN) developed using TensorFlow/Keras for binary image classification. The model classifies input images as either **Dog** or **Cat**.
This project was created as part of my deep learning portfolio to demonstrate CNN design, model training, evaluation, and deployment practices.
---
## 🧠 Model Details
- **Framework:** TensorFlow / Keras
- **Architecture:** Convolutional Neural Network (CNN)
- **Task:** Binary Image Classification
- **Classes:** Dog, Cat
- **Input Size:** 256 Γ— 256 Γ— 3
- **Epochs:** 10
- **Validation Accuracy:** ~95–96%
---
## πŸ“Š Training
The model was trained on the Kaggle Dogs vs Cats dataset.
Training included:
- Image preprocessing
- CNN feature extraction
- Binary classification using a sigmoid output layer
- Model evaluation using accuracy and loss metrics
---
## πŸš€ Usage
```python
from tensorflow.keras.models import load_model
model = load_model("cnn_model.keras")
```
---
## πŸ“ˆ Results
- Validation Accuracy: **95–96%**
- Binary Classification
- TensorFlow/Keras Implementation
---
## πŸ“š Future Improvements
- Transfer Learning (EfficientNet / ResNet50)
- Data Augmentation
- Hyperparameter Tuning
- Grad-CAM Visualization
- Streamlit Deployment
---
## πŸ‘¨β€πŸ’» Author
**Vertika**
GitHub:
https://github.com/vertika13122007-tech