Image classification with Azure Custom Vision Service and CoreML

By | June 4, 2018

In my previous blogpost I created a small PoC to show you how to work with the Azure Custom Vision Service and WinML. In this blogpost we are going to take the same model that I have trained back then, export it as a CoreML model and use it in iOS using Xamarin. I was surprised by how easy it is to do!

Export the model and convert it

Firstly, I’m going back to Azure Custom Vision Service where I previously created a project and trained a model that classifies pictures of Disney castles. More information about this can be found in my previous blogpost.

In the Performance tab, I can click the Export button and select CoreML to download a Machine Learning Model that can be used by iOS’ CoreML framework. Once downloaded, there is an additional thing we need to do in order to use it as iOS requires a compiled model (a directory ending with .mlmodelc). To compile our .mlmodel file that we just downloaded, we need to execute the following script (on your Mac):

This is what the resulting .mlmodelc folder looks like:

The resulting mlmodelc folder

Once we have done this, we can open Visual Studio and create a new Xamarin iOS project.

Add compiled model to Xamarin.iOS project

In Visual Studio, right click the Resources folder of the Xamarin.iOS project and select ‘Add Existing Folder…’. Select the *.mlmodelc folder that the CoreMLCompiler has created. Make sure that the Build Action of every file that has been included  ( *.bin, *.net, *.shape, *.weights) is set to BundleResource.Compiled model added as Resource

Initialise the model and classification request

Finally, we need to write (very little) code to use our model to do our classification.

First, we need to create an NSUrl that points to the .mlmodelc folder in the app’s resources.

Next, we need to create an MLModel from this url.

As we are working with images, we are not just going to use this model directly. Instead, we will be using the VNCoreMLModel object which is a wrapper around a Core ML model, that can be used to do Vision recognition requests.

Once we have an instance of a VNCoreMLModel object, containing our trained classification model, we can create a VNCoreMLRequest object. To create an instance of a VNCoreMLRequest object, we pass in the VNCoreMLModel and a completion handler. Fast forward: later on in code we will reference this request object. When the classification is done, the completion handler delegate will be called where we are able to read the result of the classification.

Picking an image and showing it on screen

In the app, there is a button which allows the user to select an image from the phone’s album. Once an image is selected, it is shown on screen and a button appears that will do the classification of the chosen image. Not going into detail here, as this ‘generic’ code has nothing to do with the topic itself.

Doing the image classification

Once the user has selected an image, a button will appear. When clicking this button, the chosen image will be classified using the Machine Learning Model that we initially created in Azure, downloaded, compiled and referenced in our app.

First, we create a VNImageRequestHandler. This is an object that process one or more image analysis requests on a single image. We pass in the image selected by the user.

Next, we can call the Perform method on this object, passing in the analysis we want to do. Earlier we initialised the ClassificationRequest member which has a reference to our Machine Learning Model and a completion handler that get’s triggered when the classification is done. So we pass in this ClassificationRequest.

Once the classification is done, we can get the result in the completion handler delegate of the ClassificationRequest:

We can get the classification by calling the generic GetResults method on the request object, passing in VNClassificationObservation as the type parameter (because classifications is what we expect).

We take the first result from this array, as this is the one with the highest confidence. All that is left to do, is show this to the user! Done! 😎

Image classification CoreML

Conclusion

It is very easy to train your image classification models in Azure, using the Custom Vision Service. Exporting these models and using them offline in combination with CoreML or WinML (as discussed in an earlier blogpost), is just peanuts! Adding artificial intelligence to your apps has never been easier! 🙂

The code can be found on GitHub.

 

 

 

 

One thought on “Image classification with Azure Custom Vision Service and CoreML

Leave a Reply