A photograph of a fruit stall in a market with a woman buying fruit. Items in the image have a tag label next to them with the fruit or object name. This symbolizes how the SageMaker image classification algorithm works.

Image Classification Algorithm

ByMichael Stainsbury Last updated date:

The SageMaker Image Classification algorithm can apply multiple labels to an image depending on what objects are identified. Objects are either identified, or not, there are no probability scores.

Attributes

Problem attributeDescription
Data types and formatImage
Learning paradigm or domainImage Processing, Supervised
Problem typeImage and multi-label classification
Use case examplesLabel/tag an image based on the content of the image

Training

The recommended format for training data for the image classification algorithm is the Apache MxNet recordIO format, although it will also accept jpeg and png. Incremental training can be used to speed up training. This is where a previously trained model is used for the training data.

Model artifacts and inference

DescriptionArtifacts
Learning paradigmSupervised
Request formatapplication/x-image, jpeg, png
Resultapplication/x-image

Processing environment

Only GPU instances are recommended for the Image Classification algorithm. Multiple GPU instances and single instances with multiple GPUs can be used.

AWS Innovate | Intro to Deep Learning: Building an Image Classifier on Amazon SageMaker

This is a one hour and four minute video featuring the Image Classification algorithm, by Gabe Hollombe, from AWS.

https://www.youtube.com/watch?v=KCzgR7eQ3PY

Credits

Woman and vegetable store photo by Clem Onojeghuo on Unsplash

Similar Posts

SageMaker text processing algorithms

SageMaker text processing algorithms

ByMichael Stainsbury

There are four SageMaker text processing algorithms: BlazingText, LDA, NTM and Sequence-to-sequence. BlazingText converts text to numeric vectors. LDA and NTM identify topics in text documents and Sequence-to-sequence provides machine translation of languages. Each algorithm has it’s own section and embedded video. These revision notes are part of subdomain 3.2 Select the appropriate model(s) for…

SageMaker supervised algorithms

SageMaker supervised algorithms

ByMichael Stainsbury

There are five SageMaker supervised algorithms for tabular data. DeepAR Forecasting uses Deep Learning for financial forecasting. Linear Learner is good for regression problems. Factorization Machines can be used for the same purpose, but can handle data with gaps and holes better. K-Nearest Neighbor is good at categorising data. XGBoost can predict if an item…

Supervised Learning for Machine Learning

Supervised Learning for Machine Learning

ByMichael Stainsbury

What is Supervised Learning? For Supervised Learning you need labeled training data. In Supervised Learning we provide data that has already been identified and therefore labeled, as being what we are looking for. Once the Machine Learning model has been trained it can then be presented with real unknown data to which the Machine Learning…

Latent Dirichlet Allocation Algorithm

Latent Dirichlet Allocation Algorithm

ByMichael Stainsbury

SageMaker Latent Dirichlet Allocation algorithm (LDA) is an Unsupervised Learning algorithm that groups words in a document into topics. The topics are found by a probability distribution of all the words in a document. LDA can be used to discover topics shared by documents within a text corpus. The number of topics is specified by…