Instructions to use Vertika-1312/dog_vs_cat_classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Keras
How to use Vertika-1312/dog_vs_cat_classifier with Keras:
# Available backend options are: "jax", "torch", "tensorflow". import os os.environ["KERAS_BACKEND"] = "jax" import keras model = keras.saving.load_model("hf://Vertika-1312/dog_vs_cat_classifier") - Notebooks
- Google Colab
- Kaggle
πΆπ± 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
- Downloads last month
- 18