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
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-ic-bit-m-r101x1-imagenet21k-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-bit-m-r101x3-ilsvrc2012-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-bit-m-r101x3-imagenet21k-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-bit-m-r152x4-ilsvrc2012 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-bit-m-r152x4-imagenet21k |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-bit-m-r50x1-ilsvrc2012-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-bit-m-r50x1-imagenet21k-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-bit-m-r50x3-ilsvrc2012-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-bit-m-r50x3-imagenet21k-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-bit-s-r101x1-ilsvrc2012-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-bit-s-r101x3-ilsvrc2012-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-bit-s-r152x4-ilsvrc2012 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-bit-s-r50x1-ilsvrc2012-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-bit-s-r50x3-ilsvrc2012-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-cait-m36-384 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-cait-m48-448 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-cait-s24-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-cait-s24-384 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-cait-s36-384 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-cait-xs24-384 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-cait-xxs24-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-cait-xxs24-384 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-cait-xxs36-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-cait-xxs36-384 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-deit-base-distilled-patch16-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-deit-base-distilled-patch16-384 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-deit-base-patch16-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-deit-base-patch16-384 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-deit-small-distilled-patch16-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-deit-small-patch16-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-deit-tiny-distilled-patch16-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-deit-tiny-patch16-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-b0-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-b1-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-b2-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-b3-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-b4-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-b5-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-b6-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-b7-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-lite0-classification-2 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-lite1-classification-2 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-lite2-classification-2 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-lite3-classification-2 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-lite4-classification-2 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet1k-b0 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet1k-b1 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet1k-b2 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet1k-b3 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet1k-l |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet1k-m |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet1k-s |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-b0 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-b1 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-b2 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-b3 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b0 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b1 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b2 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-b3 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-l |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-m |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-s |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-ft1k-xl |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-l |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-m |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-s |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-efficientnet-v2-imagenet21k-xl |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-inception-resnet-v2-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-inception-v1-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-inception-v2-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-inception-v3-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-025-128-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-025-160-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-025-192-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-025-224-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-050-128-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-050-160-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-050-192-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-050-224-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-075-128-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-075-160-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-075-192-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-075-224-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-100-128-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-100-160-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-100-192-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v1-100-224-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-035-128 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-035-160 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-035-192 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-035-224-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-035-96 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-050-128 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-050-160 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-050-192 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-050-224-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-050-96 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-075-128 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-075-160 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-075-192 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-075-224-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-075-96 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-100-160 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-100-192 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-100-224-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-100-96 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-130-224-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v2-140-224-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v3-large-075-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v3-large-100-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v3-small-075-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-mobilenet-v3-small-100-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-nasnet-large |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-nasnet-mobile |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-pnasnet-large |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-resnet-v1-101-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-resnet-v1-152-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-resnet-v1-50-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-resnet-v2-101-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-resnet-v2-152-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-imagenet-resnet-v2-50-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-resnet-50-classification-1 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-swin-base-patch4-window12-384 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-swin-base-patch4-window7-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-swin-large-patch4-window12-384 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-swin-large-patch4-window7-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-swin-s3-base-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-swin-s3-small-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-swin-s3-tiny-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-swin-small-patch4-window7-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-swin-tiny-patch4-window7-224 |
True |
1.0.0 |
2.80.0 |
|
tensorflow-ic-tf2-preview-inception-v3-classification-4 |
True |
3.0.0 |
2.80.0 |
|
tensorflow-ic-tf2-preview-mobilenet-v2-classification-4 |
True |
3.0.0 |
2.80.0 |