Skip to content

Latest commit

 

History

History
43 lines (32 loc) · 841 Bytes

File metadata and controls

43 lines (32 loc) · 841 Bytes

Dog vs Cat Classifier

This project is a Convolutional Neural Network (CNN) built using Keras to classify images of dogs and cats.

Installation

  1. Clone the repository:

    git clone <repository-url>
  2. Navigate to the project directory:

    cd <project-directory>
  3. Create and activate a virtual environment:

    python -m venv .venv
    .\.venv\Scripts\activate  # On Windows
    source .venv/bin/activate  # On macOS/Linux
  4. Install dependencies:

    pip install -r requirements.txt

Usage

  1. Train the model:

    python train_model.py
  2. Predict using the trained model:

    python predict_image.py --image_path <path-to-image>

Files

  • model.h5: Saved model file.