Modelli TensorFlow - Amazon SageMaker AI

Modelli TensorFlow

I seguenti modelli preaddestrati sono disponibili per l'uso per il trasferimento dell'addestramento con l'algoritmo Rilevamento degli oggetti - TensorFlow.

I seguenti modelli variano in modo significativo in termini di dimensioni, numero di parametri del modello, tempo di addestramento e latenza di inferenza per un determinato set di dati. Il modello migliore per il tuo caso d'uso dipende dalla complessità del set di dati di ottimizzazione e dai requisiti richiesti in termini di tempo di addestramento, latenza di inferenza o precisione del modello.

Nome modello model_id Origine

ResNet50 V1 FPN 640

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

Link a TensorFlow Model Garden

EfficientDET D0 512

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

Link a TensorFlow Model Garden

EfficientDet D1 640

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

Link a TensorFlow Model Garden

EfficientDet D2 768

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

Link a TensorFlow Model Garden

EfficientDET D3 896

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

Link a TensorFlow Model Garden

MobileNet V1 FPN 640

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

Link a TensorFlow Model Garden

MobileNet V2 FPNLite 320

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

Link a TensorFlow Model Garden

MobileNet V2 FPNLite 640

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

Link a TensorFlow Model Garden

ResNet50 V1 FPN 1024

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

Link a TensorFlow Model Garden

ResNet101 V1 FPN 640

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

Link a TensorFlow Model Garden

ResNet101 V1 FPN 1024

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

Link a TensorFlow Model Garden

ResNet152 V1 FPN 640

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

Link a TensorFlow Model Garden

ResNet152 V1 FPN 1024

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

Link a TensorFlow Model Garden