Pet Image Classifier

Setup Instructions

1. Create a Virtual Environment

python3 -m venv .venv

2. Activate the Virtual Environment

On Linux/macOS:

source .venv/bin/activate

On Windows:

.venv\Scripts\activate

3. Install Dependencies

pip install --upgrade pip
pip install -r requirements.txt

4. Run the Project

Navigate to the data directory and run:

cd data
python classify_images.py --dir <image_directory> --arch <model_name> --dogfile dognames.txt

Example:

python classify_images.py --dir pet_images --arch resnet --dogfile dognames.txt

5. Deactivate Virtual Environment

When finished:

deactivate