🐱 Cat vs Dog Image Classifier

A TensorFlow CNN model for binary image classification.

Classes

  • Cat
  • Dog

Architecture

  • Conv2D
  • MaxPooling2D
  • Dense
  • Dropout

Input Size

150x150 RGB

Accuracy

Validation Accuracy: 95%

Installation

pip install tensorflow pillow numpy

Download

from huggingface_hub import hf_hub_download

model_path = hf_hub_download(
    repo_id="PatelDhvanit/Cat-Dog-Classifier",
    filename="cat_dog_model.h5"
)

Prediction

import tensorflow as tf

model=tf.keras.models.load_model(model_path)

Author

Dhvanit Patel

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support