Object Detection - TensorFlow

This is a supervised object detection 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 Object Detection 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-od-centernet-hourglass-1024x1024-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-centernet-hourglass-1024x1024-kpts-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-centernet-hourglass-512x512-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-centernet-hourglass-512x512-kpts-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-centernet-resnet101v1-fpn-512x512-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-centernet-resnet50v1-fpn-512x512-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-centernet-resnet50v1-fpn-512x512-kpts-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-centernet-resnet50v2-512x512-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-centernet-resnet50v2-512x512-kpts-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-efficientdet-d0-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-efficientdet-d1-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-efficientdet-d2-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-efficientdet-d3-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-efficientdet-d4-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-efficientdet-d5-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-faster-rcnn-inception-resnet-v2-1024x1024-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-faster-rcnn-inception-resnet-v2-640x640-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-faster-rcnn-resnet101-v1-1024x1024-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-faster-rcnn-resnet101-v1-640x640-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-faster-rcnn-resnet101-v1-800x1333-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-faster-rcnn-resnet152-v1-1024x1024-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-faster-rcnn-resnet152-v1-640x640-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-faster-rcnn-resnet152-v1-800x1333-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-faster-rcnn-resnet50-v1-1024x1024-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-faster-rcnn-resnet50-v1-640x640-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-faster-rcnn-resnet50-v1-800x1333-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-retinanet-resnet101-v1-fpn-1024x1024-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-retinanet-resnet101-v1-fpn-640x640-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-retinanet-resnet152-v1-fpn-1024x1024-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-retinanet-resnet152-v1-fpn-640x640-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-retinanet-resnet50-v1-fpn-1024x1024-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-retinanet-resnet50-v1-fpn-640x640-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-ssd-mobilenet-v1-fpn-640x640-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-ssd-mobilenet-v2-2

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-ssd-mobilenet-v2-fpnlite-320x320-1

False

2.0.2

2.80.0

Tensorflow Hub

tensorflow-od-ssd-mobilenet-v2-fpnlite-640x640-1

False

2.0.2

2.80.0

Tensorflow Hub