🐢🐱 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

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

Downloads last month
18
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support