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