Train a machine learning model to classify images.
- Create ML
- Create MLUI
An image classifier is a machine learning model that’s been trained to recognize images. When you give it an image, it responds with a label for that image.
You train an image classifier by showing it lots of examples of images you’ve already labeled. For example, you can train an image classifier to recognize safari animals by showing it a variety of photos of elephants, giraffes, lions, and so on.
Prepare Your Data
Start by preparing the data that you’ll use to train and evaluate the classifier. Create a training data set from about 80% of the images you have for each label. Create a testing data set from the remaining images. Make sure that any given image appears in only one of these two sets.
Next, create a folder called
Training Data, and another called
Testing Data. In each folder, create subfolders using your labels as names. Then sort the images into the appropriate subfolders for each data set.
The exact label strings aren’t important, as long as they make sense to you. For example, you might use the label
Cheetah for all the images of cheetahs. You don’t have to name the image files in any particular way or add metadata to them. You only need to put them into the folder with the right label.
Use at least 10 images per label for the training set, but more is always better. Also, balance the number of images for each label. For example, don’t use 10 images for
Cheetah and 1000 images for
The images can be in any format whose uniform type identifer conforms to
public. This includes common formats like JPEG and PNG. The images don’t have to be the same size as each other, nor do they have to be any particular size, although it’s best to use images that are at least 299x299 pixels. If possible, train with images collected in a way that’s similar to how images will be collected for prediction.
Provide images with variety. For example, use images that show animals from many different angles and in different lighting conditions. A classifier trained on nearly identical images for a given label tends to have poorer performance than one trained on a more diverse image set.
Show an Image Classifier Builder in a Playground
With your data ready, make a new Xcode playground with a macOS target. Use the playground to create an
MLImage instance and show it in the live view:
Show the assistant editor in Xcode, and then run the playground. When you do this, the live view presents an image classifier:
Train the Image Classifier
Training Data folder from Finder onto the indicated location in the live view. When you do this, the training process starts and the image classifier shows its progress:
As a part of the training process, the image classifier automatically splits your training data into a training set and a validation set. These both affect training, but in different ways. Because the split is done randomly, you might get a different result each time you train the model.
When training finishes, the live view shows training and validation accuracies. These report how well the trained model classifies images from the corresponding sets. Because the model trained on these images, it typically does a good job classifying them.
Evaluate the Classifier’s Performance
Next, evaluate your trained model’s performance by testing it with images it’s never seen before. To do this, use the test data set that you created before you started training. Drag the
Test Data folder into the live view, just as you did with the training data.
The model processes all of the images, making predictions for each. Because this is labeled data, the model can check its own predictions. It then adds the overall evaluation accuracy as the final metric in the UI.
If the evaluation performance isn’t good enough, you may need to retrain with more data, or alter some other training configuration. For information about how to do more detailed model evaluation, as well as strategies for improving model performance, see Improving Your Model’s Accuracy.
Save the Core ML Model
When your model performs well enough, save it so that you can use it in your app.
Give the classifier a meaningful name. Change the default
Image to be
Animal by changing the name in the UI. You can also add more information about the model, like the author and a short description. Click the disclosure triangle to reveal these metadata fields and fill in the details.
Click Save. The model writes a file in
.mlmodel format into the directory specified by the
Add the Model to Your App
Now add the trained model to an existing Core ML enabled app. You can use this model to replace the one that comes with the Classifying Images with Vision and Core ML sample code project. If you do this, the sample app works exactly the same as before, except that it recognizes and classifies animals according to the labels you’ve defined.
Open the sample code project in Xcode and drag your model file into the navigation pane. Once the model is part of the project, Xcode shows you the model metadata, along with other features, like the model class.
To use the new model in code, you change only one line. The MobileNet model that comes with the project is instantiated in exactly one place in the
Change this one line to use the new model class instead:
The models are interchangeable because both take an image as input and output a label. After your substitution, the sample app classifies images just as before, except it uses your model and its associated labels.