...

/

Image Classifier Using SDK - Uploading and Training the Model

Image Classifier Using SDK - Uploading and Training the Model

Learn to build an Image Classifier using the Azure Custom Vision SDK for Python.

Let’s continue the implementation part. Just to recap, in the previous lesson, we’ve created the training and prediction client objects. We’ve also created a project and added the two image tags that we’re interested in.

Now, we’re going to upload the images with their tags to our Custom Vision project and then we’ll train our custom model.

Uploading the images with their tags

Now let’s upload the images and their corresponding tags to the Custom Vision project.

Press + to interact
base_image_location = "CourseAssets/ImageClassification/Images"
print("Adding images...")
image_list = []
for image_num in range(1, 11):
file_name = "hemlock_{}.jpg".format(image_num)
with open(base_image_location + "/Hemlock/" + file_name, "rb") as hemlock_image:
image_list.append(
ImageFileCreateEntry(
name = file_name,
contents = hemlock_image.read(),
tag_ids= [hemlock_tag.id]
)
)
for image_num in range(1, 11):
file_name = "japanese_cherry_{}.jpg".format(image_num)
with open(base_image_location + "/Japanese_Cherry/" + file_name, "rb") as cherry_image:
image_list.append(
ImageFileCreateEntry(
name = file_name,
contents = cherry_image.read(),
tag_ids = [cherry_tag.id]
)
)
upload_result = trainer.create_images_from_files(
project.id,
ImageFileCreateBatch(images = image_list)
)
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("Imaged Added Successfully!!")
  • In line 1, we define the location of the “Images” folder present in our working directory.

  • In line 5, we define a list named image_list that will ...