shangfuu / Image_classifier

Image classify a photo is cat or dog.

Image_classifier

Classify photos of Dogs and Cats.

Dataset: https://www.microsoft.com/en-us/download/details.aspx?id=54765

Data

• Organise the dataset as follows:

Data
    pickles
        features.pickle
        labels.pickle
    models
        64x3-CNN.model
testData
    (images)    
PetImages
    Cat
        (images)
    Dog
        (images)

Install

• Type pip install -r requirements.txt to install all packages.
• Please check config.py for following steps.

Prepare dataset

Type python prepare.py to make pickle files for your PetImages.

Training

Type python train.py to train simple model/ Type python train.py -m hard to train model with more accuracy one.

Testing

Type python app.py to test your cat / dog images in testData folder.