TensorFlow Models
The following pretrained models are available to use for transfer learning with the Object Detection - TensorFlow algorithm.
The following models vary significantly in size, number of model parameters, training time, and inference latency for any given dataset. The best model for your use case depends on the complexity of your fine-tuning dataset and any requirements that you have on training time, inference latency, or model accuracy.
Model Name | model_id |
Source |
---|---|---|
ResNet50 V1 FPN 640 |
|
|
EfficientDet D0 512 |
|
|
EfficientDet D1 640 |
|
|
EfficientDet D2 768 |
|
|
EfficientDet D3 896 |
|
|
MobileNet V1 FPN 640 |
|
|
MobileNet V2 FPNLite 320 |
|
|
MobileNet V2 FPNLite 640 |
|
|
ResNet50 V1 FPN 1024 |
|
|
ResNet101 V1 FPN 640 |
|
|
ResNet101 V1 FPN 1024 |
|
|
ResNet152 V1 FPN 640 |
|
|
ResNet152 V1 FPN 1024 |
|