Image Classification - TensorFlow

This is a supervised image clasification algorithm which supports fine-tuning of many pre-trained models available in Tensorflow Hub. The following sample notebook demonstrates how to use the Sagemaker Python SDK for Image Classification for using these algorithms.

For detailed documentation please refer Use Built-in Algorithms with Pre-trained Models in SageMaker Python SDK

Available Models

Model ID

Fine Tunable?

Latest Version

Min SDK Version

Source

tensorflow-ic-bit-m-r101x1-ilsvrc2012-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-bit-m-r101x1-imagenet21k-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-bit-m-r101x3-ilsvrc2012-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-bit-m-r101x3-imagenet21k-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-bit-m-r152x4-ilsvrc2012

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-bit-m-r152x4-imagenet21k

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-bit-m-r50x1-ilsvrc2012-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-bit-m-r50x1-imagenet21k-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-bit-m-r50x3-ilsvrc2012-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-bit-m-r50x3-imagenet21k-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-bit-s-r101x1-ilsvrc2012-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-bit-s-r101x3-ilsvrc2012-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-bit-s-r152x4-ilsvrc2012

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-bit-s-r50x1-ilsvrc2012-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-bit-s-r50x3-ilsvrc2012-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-cait-m36-384

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-cait-m48-448

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-cait-s24-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-cait-s24-384

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-cait-s36-384

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-cait-xs24-384

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-cait-xxs24-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-cait-xxs24-384

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-cait-xxs36-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-cait-xxs36-384

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-deit-base-distilled-patch16-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-deit-base-distilled-patch16-384

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-deit-base-patch16-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-deit-base-patch16-384

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-deit-small-distilled-patch16-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-deit-small-patch16-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-deit-tiny-distilled-patch16-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-deit-tiny-patch16-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-b0-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-b1-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-b2-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-b3-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-b4-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-b5-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-b6-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-b7-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-lite0-classification-2

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-lite1-classification-2

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-lite2-classification-2

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-lite3-classification-2

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-lite4-classification-2

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet1k-b0

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet1k-b1

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet1k-b2

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet1k-b3

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet1k-l

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet1k-m

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet1k-s

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-b0

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-b1

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-b2

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-b3

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b0

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b1

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b2

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b3

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-l

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-m

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-s

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-xl

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-l

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-m

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-s

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-efficientnet-v2-imagenet21k-xl

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-inception-resnet-v2-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-inception-v1-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-inception-v2-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-inception-v3-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-025-128-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-025-160-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-025-192-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-025-224-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-050-128-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-050-160-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-050-192-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-050-224-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-075-128-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-075-160-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-075-192-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-075-224-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-100-128-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-100-160-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-100-192-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v1-100-224-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-035-128

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-035-160

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-035-192

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-035-224-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-035-96

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-050-128

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-050-160

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-050-192

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-050-224-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-050-96

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-075-128

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-075-160

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-075-192

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-075-224-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-075-96

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-100-160

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-100-192

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-100-224-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-100-96

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-130-224-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v2-140-224-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v3-large-075-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v3-large-100-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v3-small-075-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-mobilenet-v3-small-100-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-nasnet-large

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-nasnet-mobile

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-pnasnet-large

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-resnet-v1-101-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-resnet-v1-152-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-resnet-v1-50-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-resnet-v2-101-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-resnet-v2-152-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-imagenet-resnet-v2-50-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-resnet-50-classification-1

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-swin-base-patch4-window12-384

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-swin-base-patch4-window7-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-swin-large-patch4-window12-384

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-swin-large-patch4-window7-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-swin-s3-base-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-swin-s3-small-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-swin-s3-tiny-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-swin-small-patch4-window7-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-swin-tiny-patch4-window7-224

True

1.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-tf2-preview-inception-v3-classification-4

True

3.0.0

2.80.0

Tensorflow Hub

tensorflow-ic-tf2-preview-mobilenet-v2-classification-4

True

3.0.0

2.80.0

Tensorflow Hub