TensorFlow Modèles - Amazon SageMaker

Les traductions sont fournies par des outils de traduction automatique. En cas de conflit entre le contenu d'une traduction et celui de la version originale en anglais, la version anglaise prévaudra.

TensorFlow Modèles

Les modèles préentraînés suivants peuvent être utilisés pour l'apprentissage par transfert avec l' TensorFlow algorithme de détection d'objets.

Les modèles suivants varient de manière significative par leur taille, le nombre de paramètres de modèle, la durée d'entraînement et la latence d'inférence pour n'importe quel jeu de données. Le meilleur modèle pour votre cas d'utilisation dépend de la complexité de l'affinage du jeu de données et de toutes vos exigences en matière de durée d'entraînement, de latence d'inférence ou de précision du modèle.

Nom du modèle model_id Source

ResNet50 V1 FPN 640

tensorflow-od1-ssd-resnet50-v1-fpn-640x640-coco17-tpu-8

TensorFlow Model Garden link

EfficientDet D0 512

tensorflow-od1-ssd-efficientdet-d0-512x512-coco17-tpu-8

TensorFlow Model Garden link

EfficientDet D1 640

tensorflow-od1-ssd-efficientdet-d1-640x640-coco17-tpu-8

TensorFlow Model Garden link

EfficientDet D2 768

tensorflow-od1-ssd-efficientdet-d2-768x768-coco17-tpu-8

TensorFlow Model Garden link

EfficientDet D3 896

tensorflow-od1-ssd-efficientdet-d3-896x896-coco17-tpu-32

TensorFlow Model Garden link

MobileNet V1 FPN 640

tensorflow-od1-ssd-mobilenet-v1-fpn-640x640-coco17-tpu-8

TensorFlow Model Garden link

MobileNet V2 FPN Lite 320

tensorflow-od1-ssd-mobilenet-v2-fpnlite-320x320-coco17-tpu-8

TensorFlow Model Garden link

MobileNet V2 FPN Lite 640

tensorflow-od1-ssd-mobilenet-v2-fpnlite-640x640-coco17-tpu-8

TensorFlow Model Garden link

ResNet50 V1 VPN 1024

tensorflow-od1-ssd-resnet50-v1-fpn-1024x1024-coco17-tpu-8

TensorFlow Model Garden link

ResNet101 V1 FPN 640

tensorflow-od1-ssd-resnet101-v1-fpn-640x640-coco17-tpu-8

TensorFlow Model Garden link

ResNet101 V1 FPN 1024

tensorflow-od1-ssd-resnet101-v1-fpn-1024x1024-coco17-tpu-8

TensorFlow Model Garden link

ResNet152 V1 FPN 640

tensorflow-od1-ssd-resnet152-v1-fpn-640x640-coco17-tpu-8

TensorFlow Model Garden link

ResNet152 V1 FPN 1024

tensorflow-od1-ssd-resnet152-v1-fpn-1024x1024-coco17-tpu-8

TensorFlow Model Garden link