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 |