Comment on page
Lab 1 - Train your model
Create a custom vision model using the Microsoft Custom Vision service.
The Custom Vision Endpoint must be created inside a resource group. You can use an existing resource group or create a new one.
To create a new resource group, use the following command. Replace <<resource-group-name>> with your name to use for this resource group. Replace <<location>> with the Azure region to use for this resource group.
az group create --name <resource-group-name> --location <location>
Run the command below to create a free Custom Vision Training Endpoint. Replace <<resource-group-name>> with the name you used above and use the same location. Replace <<name>> with a name for your resource like: my-custom-vision-training.
az cognitiveservices account create --name <name> --kind CustomVision.Training --sku F0 --resource-group <resource-group-name> --location <location> --yes
To get the endpoint from the Cognitive service account run the CLI command below. Replace <<name>> and <<resource-group-name>> with the names used above.
Get the API keys
az cognitiveservices account keys list --name <name> --resource-group <resource-group-name>
Get the endpoint URL
az cognitiveservices account show --name <name> --resource-group <resource-group-name> -o json --query properties.endpoint
- An endpoint URL looking like this: https://<region>.api.cognitive.microsoft.com/
- 2 keys looking like this: 06a611d19f4f4a88a03f3b552a5d2379
For creating the model we are going to use a Jupyter notebook running in Visual Studio Code.
In the terminal type:
code cv.ipynb
This will open a blank notebook in Visual Studio Code.
- A notebook has cells
- In a cells you put code.
- You can run the cell to execute code within that single cell
- In the first cell type:
print("hello world")
- Click on the play button on the left of the cell
- The first time you have to select the Kernel your notebook is using, select the top one.
- The result of the code is displayed below the cell
- To add a new cell you can click the "+code" button in the top menu or if you move your mouse below a cell.
Add every block of code in this lab in a separate cell and excecute the cell as you go.
Every Machine Learning journey starts with a question you want to have answered. For this example, you are going to answer the question: Is it a Homer or a Marge Lego figure.
Now that we know what to ask the model, we can go on to the next requirement; that is data. Our model is going to be a classification model, meaning the model will look at the picture and scores the pictures against the different classes. So, the output will be I’m 70% confident this is Homer and 1% confident that this is Marge. By taking the class with the highest score and setting a minimum threshold for the confidence score we know what is in the picture.
I have created a dataset for you with 50 pictures of a Homer Simpson Lego figure and 50 pictures of a Marge Simpsons Lego figure. I have taken the photos with a few things in mind:
- Used a lot of different backgrounds
- Took the photos from different angles.
- I made sure the only object in the photo was Homer or Marge
- The quality of the photos was consistent.
Use the Python code below to download and extract the dataset to the folder "data-train" or download the dataset and extract it manually in the folder "data-train".
import os
import urllib.request
import zipfile
# Download the dataset from Github
data_url = "https://github.com/hnky/dataset-lego-figures/raw/master/_download/simpsons-lego-dataset.zip"
data_path = "./data-train"
download_path = os.path.join(data_path,"simpsons-lego-dataset.zip")
if not os.path.exists(data_path):
os.mkdir(data_path);
urllib.request.urlretrieve(data_url, filename=download_path)
# Unzip the dataset
zip_ref = zipfile.ZipFile(download_path, 'r')
zip_ref.extractall(data_path)
zip_ref.close()
print("Data extracted in: {}".format(data_path))
os.remove(download_path)
print("Downloaded file removed: {}".format(download_path))
Start with importing the packages needed.
from azure.cognitiveservices.vision.customvision.training import CustomVisionTrainingClient
from azure.cognitiveservices.vision.customvision.training.models import ImageFileCreateEntry, ImageFileCreateBatch
from msrest.authentication import ApiKeyCredentials
import time
Next, create variables for the Custom Vision endpoint, Custom Vision training key and the location where the training images are stored.
cv_endpoint = "https://<REGION>.api.cognitive.microsoft.com"
training_key = "<INSERT KEY 1>"
training_images = "data-train"
To start with the training, we need to create a Training Client. This method takes as input the endpoint and the training key.
credentials = ApiKeyCredentials(in_headers={"Training-key": training_key})
trainer = CustomVisionTrainingClient(endpoint=cv_endpoint,credentials=credentials)
Now you are ready to create your first project. The project takes a name and domain as input, the name can be anything. The domain is a different story. You can ask for a list of all possible domains and choose the one closest to what you are trying to accomplish. For instance if you are trying to classify food you pick the domain “Food” or “Landmarks” for landmarks. Use the code below to show all domains.
for domain in trainer.get_domains():
print(domain.id, "\t", domain.name)
You might notice that some domains have the word “Compact” behind them. If this is the case it means the Azure Custom Vision Service will create a smaller model, which you will be able to export and run locally on your mobile phone or desktop.
Let’s create a new project with the domain set to “General Compact”.
project = trainer.create_project("Lego Simpsons",domain_id="0732100f-1a38-4e49-a514-c9b44c697ab5")
Next you need to create tags, these tags are the same as classes mentioned above. When you have created a few tags we can tag images with them and upload the images to the Azure Custom Vision Service.
Our images are sorted per tag/class in a folder. All the photos of Marge are in the folder named 'Marge' and all the images of Homer are in the folder named 'Homer'.
In the code below we do the following steps:
- We open the directory containing the folders with training images.
- Loop through all the directories found in this folder
- Create a new tag with the folder name
- Open the folder containing the images
- Create, for every image in that folder, an ImageFileEntry that contains the filename, file content and the tag.
- Add this ImageFileEntry to a list.
image_list = []
directories = os.listdir(training_images)
for tagName in directories:
tag = trainer.create_tag(project.id, tagName)
images = os.listdir(os.path.join(training_images,tagName))
for img in images:
with open(os.path.join(training_images,tagName,img), "rb") as image_contents:
image_list.append(ImageFileCreateEntry(name=img, contents=image_contents.read(), tag_ids=[tag.id]))
Now you have a list that contains all tagged images. So far no images have been added to the Azure Custom Vision service, only the tags have been created.
Uploading images goes in batches with a max size of 64 images per batch. Our dataset is 300+ images big, so first we need to split the list into chunks of 64 images.
def chunks(l, n):
for i in range(0, len(l), n):
yield l[i:i + n]
batchedImages = chunks(image_list, 64)
Now we have our images split in batches of 64, we can upload them batch by batch to the Azure Custom Vision Service. Note: This can take a while!
for batchOfImages in batchedImages:
upload_result = trainer.create_images_from_files(project.id, ImageFileCreateBatch(images=batchOfImages))
if not upload_result.is_batch_successful:
print("Image batch upload failed.")
for image in upload_result.images:
print("Image status: ", image.status)
else:
print("Batch uploaded successfully")
print("Done uploading")
From this point, there are only two steps remaining before you can access the model through an API endpoint. First you need to train the model and finally you must publish the model, so it is accessible through a prediction API. The training can take a while, so you can create a while loop after the train request that checks the status of the model training every second.
print ("Training...")
iteration = trainer.train_project(project.id)
while (iteration.status != "Completed"):
iteration = trainer.get_iteration(project.id, iteration.id)
print ("Training status: " + iteration.status)
print ("Waiting 25 seconds...")
time.sleep(25)
print("Training complete")
Now you have successfully trained your model!
To test our model we are going to export our model in the ONNX format, download the model and run it locally.
Use the code below to start the export. Creating an export can take a while depending on the size of your model and the platform of choice.
platform = "ONNX"
flavor = "ONNX12"
iteration_id = iteration.id
project_id = project.id
export = trainer.export_iteration(project_id, iteration_id , platform, flavor, raw=False)
The code below checks if the export is done every 10 seconds
while (export.status == "Exporting"):
print ("Waiting 10 seconds...")
time.sleep(10)
exports = trainer.get_exports(project.id, iteration_id)
# Locate the export for this iteration and check its status
for e in exports:
if e.platform == export.platform and e.flavor == export.flavor:
export = e
break
print("Export status is: ", export.status)
print("Export: done")
When the export is done we can download the model in a zip file and extract the zip file in the ./model directory.
import os
import requests
import zipfile
if export.status == "Done":
# Success, now we can download it
export_file = requests.get(export.download_uri)
with open("export.zip", "wb") as file:
file.write(export_file.content)
# Unzip the downloaded export
if not os.path.exists("./model"):
os.mkdir("./model");
zip_ref = zipfile.ZipFile("export.zip", 'r')
zip_ref.extractall("./model")
zip_ref.close()
print("Data extracted in: ./model")
To test the model we need a images that the model never has seen. This code downloads the training set and unzips the set in the folder /data-test
# Download the dataset from Github
data_url = "https://github.com/hnky/dataset-lego-figures/raw/master/_download/test-images.zip"
data_path = "./data-test"
download_path = os.path.join(data_path,"test-images.zip")
if not os.path.exists(data_path):
os.mkdir(data_path);
urllib.request.urlretrieve(data_url, filename=download_path)
# Unzip the dataset
zip_ref = zipfile.ZipFile(download_path, 'r')
zip_ref.extractall(data_path)
zip_ref.close()
print("Data extracted in: {}".format(data_path))
os.remove(download_path)
print("Downloaded file removed: {}".format(download_path))
Finally we can see if our model works. The code below:
1. Gets all the images in the data-test directory
2. Loads the model
3. Loops over the array with test images
4. Opens and scales the image the right format for the model
5. Runs it through the model
6. Print the prediction to the console
import onnxruntime as nxrun
import numpy as np
import PIL
from PIL import Image
training_images = "./data-test"
model_path = "./model/model.onnx"
sess = nxrun.InferenceSession(model_path)
testimages = os.listdir(training_images)
for image_filepath in testimages[0:5]:
image = PIL.Image.open(os.path.join(training_images,image_filepath)).resize([224,224])
input_array = np.array(image, dtype=np.float32)[np.newaxis, :, :, :]
input_array = input_array.transpose((0, 3, 1, 2))[:, (2, 1, 0), :, :]
input_name = sess.get_inputs()[0].name
outputs = sess.run(None, {input_name: input_array.astype(np.float32)})
print("Image:", image_filepath)
print("Label: " + outputs[0][0][0])
print("Score: " + str(outputs[1][0][outputs[0][0][0]]))
print("--")
The end result
Great work!
You have created your specialized Simpsons classification model using the Azure Custom Vision Service.
- You created an Azure Resource group containing an Azure Custom Vision service training and prediction endpoint
- You have created a new Project
- In that project you have created tags
- You have uploaded images in batches of 64 and tagged them
- You have trained an iteration of your model
- You have exported your model in the ONNX format
- You have tested the model on a test dataset