Skip to content
Go to file
Cannot retrieve contributors at this time
82 lines (57 sloc) 2.9 KB

Image Classification

Given an image, the goal of an image classifier is to assign it to one of a pre-determined number of labels. Deep learning methods have recently been shown to give incredible results on this challenging problem. Yet this comes at the cost of extreme sensitivity to model hyper-parameters and long training time. This means that one can spend months testing different model configurations, much too long to be worth the effort. However, the image classifier in Turi Create is designed to minimize these pains, and making it possible to easily create a high quality image classifier model.

Loading Data

The Kaggle Cats and Dogs Dataset provides labeled cat and dog images.1 After downloading and decompressing the dataset, navigate to the main kagglecatsanddogs folder, which contains a PetImages subfolder.

import turicreate as tc

# Load images (Note:'Not a JPEG file' errors are warnings, meaning those files will be skipped)
data = tc.image_analysis.load_images('PetImages', with_path=True)

# From the path-name, create a label column
data['label'] = data['path'].apply(lambda path: 'dog' if '/Dog' in path else 'cat')

# Save the data for future use'cats-dogs.sframe')

# Explore interactively

Introductory Example

The task is to predict if a picture is a cat or a dog. Let’s explore the use of the image classifier on the Cats vs. Dogs dataset.

import turicreate as tc

# Load the data
data =  tc.SFrame('cats-dogs.sframe')

# Make a train-test split
train_data, test_data = data.random_split(0.8)

# Create the model
model = tc.image_classifier.create(train_data, target='label')

# Save predictions to an SArray
predictions = model.predict(test_data)

# Evaluate the model and print the results
metrics = model.evaluate(test_data)

# Save the model for later use in Turi Create'cats-dogs.model')

# Export for use in Core ML

Here are some predictions on our own favorite cats and dogs:

new_cats_dogs['predictions'] = model.predict(new_cats_dogs)

Image classifier predictions

Advanced Usage

Refer to the following chapters for:

  • Advanced usage options including the use of GPUs and deployment to device.
  • Technical details on how the image classifier works.

In addition, the following chapters contain more information on how to use classifiers:

You can’t perform that action at this time.