Object Detection - TensorFlow — sagemaker 2.199.0 documentation (original) (raw)

This is a supervised object detection algorithm which supports fine-tuning of many pre-trained models available in Tensorflow Hub. The followingsample notebookdemonstrates 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 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-centernet-hourglass-1024x1024-kpts-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-centernet-hourglass-512x512-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-centernet-hourglass-512x512-kpts-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-centernet-resnet101v1-fpn-512x512-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-centernet-resnet50v1-fpn-512x512-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-centernet-resnet50v1-fpn-512x512-kpts-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-centernet-resnet50v2-512x512-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-centernet-resnet50v2-512x512-kpts-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-efficientdet-d0-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-efficientdet-d1-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-efficientdet-d2-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-efficientdet-d3-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-efficientdet-d4-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-efficientdet-d5-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-faster-rcnn-inception-resnet-v2-1024x1024-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-faster-rcnn-inception-resnet-v2-640x640-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-faster-rcnn-resnet101-v1-1024x1024-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-faster-rcnn-resnet101-v1-640x640-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-faster-rcnn-resnet101-v1-800x1333-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-faster-rcnn-resnet152-v1-1024x1024-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-faster-rcnn-resnet152-v1-640x640-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-faster-rcnn-resnet152-v1-800x1333-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-faster-rcnn-resnet50-v1-1024x1024-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-faster-rcnn-resnet50-v1-640x640-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-faster-rcnn-resnet50-v1-800x1333-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-retinanet-resnet101-v1-fpn-1024x1024-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-retinanet-resnet101-v1-fpn-640x640-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-retinanet-resnet152-v1-fpn-1024x1024-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-retinanet-resnet152-v1-fpn-640x640-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-retinanet-resnet50-v1-fpn-1024x1024-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-retinanet-resnet50-v1-fpn-640x640-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-ssd-mobilenet-v1-fpn-640x640-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-ssd-mobilenet-v2-2 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-ssd-mobilenet-v2-fpnlite-320x320-1 False 3.0.0 2.189.0 Tensorflow Hub
tensorflow-od-ssd-mobilenet-v2-fpnlite-640x640-1 False 3.0.0 2.189.0 Tensorflow Hub