The following pretrained models are available to use for transfer learning with the Image Classification - 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 |
|---|---|---|
| MobileNet V2 1.00 224 | tensorflow-ic-imagenet-mobilenet-v2-100-224-classification-4 |
TensorFlow Hub link |
| MobileNet V2 0.75 224 | tensorflow-ic-imagenet-mobilenet-v2-075-224-classification-4 |
TensorFlow Hub link |
| MobileNet V2 0.50 224 | tensorflow-ic-imagenet-mobilenet-v2-050-224-classification-4 |
TensorFlow Hub link |
| MobileNet V2 0.35 224 | tensorflow-ic-imagenet-mobilenet-v2-035-224-classification-4 |
TensorFlow Hub link |
| MobileNet V2 1.40 224 | tensorflow-ic-imagenet-mobilenet-v2-140-224-classification-4 |
TensorFlow Hub link |
| MobileNet V2 1.30 224 | tensorflow-ic-imagenet-mobilenet-v2-130-224-classification-4 |
TensorFlow Hub link |
| MobileNet V2 | tensorflow-ic-tf2-preview-mobilenet-v2-classification-4 |
TensorFlow Hub link |
| Inception V3 | tensorflow-ic-imagenet-inception-v3-classification-4 |
TensorFlow Hub link |
| Inception V2 | tensorflow-ic-imagenet-inception-v2-classification-4 |
TensorFlow Hub link |
| Inception V1 | tensorflow-ic-imagenet-inception-v1-classification-4 |
TensorFlow Hub link |
| Inception V3 Preview | tensorflow-ic-tf2-preview-inception-v3-classification-4 |
TensorFlow Hub link |
| Inception ResNet V2 | tensorflow-ic-imagenet-inception-resnet-v2-classification-4 |
TensorFlow Hub link |
| ResNet V2 50 | tensorflow-ic-imagenet-resnet-v2-50-classification-4 |
TensorFlow Hub link |
| ResNet V2 101 | tensorflow-ic-imagenet-resnet-v2-101-classification-4 |
TensorFlow Hub link |
| ResNet V2 152 | tensorflow-ic-imagenet-resnet-v2-152-classification-4 |
TensorFlow Hub link |
| ResNet V1 50 | tensorflow-ic-imagenet-resnet-v1-50-classification-4 |
TensorFlow Hub link |
| ResNet V1 101 | tensorflow-ic-imagenet-resnet-v1-101-classification-4 |
TensorFlow Hub link |
| ResNet V1 152 | tensorflow-ic-imagenet-resnet-v1-152-classification-4 |
TensorFlow Hub link |
| ResNet 50 | tensorflow-ic-imagenet-resnet-50-classification-4 |
TensorFlow Hub link |
| EfficientNet B0 | tensorflow-ic-efficientnet-b0-classification-1 |
TensorFlow Hub link |
| EfficientNet B1 | tensorflow-ic-efficientnet-b1-classification-1 |
TensorFlow Hub link |
| EfficientNet B2 | tensorflow-ic-efficientnet-b2-classification-1 |
TensorFlow Hub link |
| EfficientNet B3 | tensorflow-ic-efficientnet-b3-classification-1 |
TensorFlow Hub link |
| EfficientNet B4 | tensorflow-ic-efficientnet-b4-classification-1 |
TensorFlow Hub link |
| EfficientNet B5 | tensorflow-ic-efficientnet-b5-classification-1 |
TensorFlow Hub link |
| EfficientNet B6 | tensorflow-ic-efficientnet-b6-classification-1 |
TensorFlow Hub link |
| EfficientNet B7 | tensorflow-ic-efficientnet-b7-classification-1 |
TensorFlow Hub link |
| EfficientNet B0 Lite | tensorflow-ic-efficientnet-lite0-classification-2 |
TensorFlow Hub link |
| EfficientNet B1 Lite | tensorflow-ic-efficientnet-lite1-classification-2 |
TensorFlow Hub link |
| EfficientNet B2 Lite | tensorflow-ic-efficientnet-lite2-classification-2 |
TensorFlow Hub link |
| EfficientNet B3 Lite | tensorflow-ic-efficientnet-lite3-classification-2 |
TensorFlow Hub link |
| EfficientNet B4 Lite | tensorflow-ic-efficientnet-lite4-classification-2 |
TensorFlow Hub link |
| MobileNet V1 1.00 224 | tensorflow-ic-imagenet-mobilenet-v1-100-224-classification-4 |
TensorFlow Hub link |
| MobileNet V1 1.00 192 | tensorflow-ic-imagenet-mobilenet-v1-100-192-classification-4 |
TensorFlow Hub link |
| MobileNet V1 1.00 160 | tensorflow-ic-imagenet-mobilenet-v1-100-160-classification-4 |
TensorFlow Hub link |
| MobileNet V1 1.00 128 | tensorflow-ic-imagenet-mobilenet-v1-100-128-classification-4 |
TensorFlow Hub link |
| MobileNet V1 0.75 224 | tensorflow-ic-imagenet-mobilenet-v1-075-224-classification-4 |
TensorFlow Hub link |
| MobileNet V1 0.75 192 | tensorflow-ic-imagenet-mobilenet-v1-075-192-classification-4 |
TensorFlow Hub link |
| MobileNet V1 0.75 160 | tensorflow-ic-imagenet-mobilenet-v1-075-160-classification-4 |
TensorFlow Hub link |
| MobileNet V1 0.75 128 | tensorflow-ic-imagenet-mobilenet-v1-075-128-classification-4 |
TensorFlow Hub link |
| MobileNet V1 0.50 224 | tensorflow-ic-imagenet-mobilenet-v1-050-224-classification-4 |
TensorFlow Hub link |
| MobileNet V1 0.50 192 | tensorflow-ic-imagenet-mobilenet-v1-050-192-classification-4 |
TensorFlow Hub link |
| MobileNet V1 1.00 160 | tensorflow-ic-imagenet-mobilenet-v1-050-160-classification-4 |
TensorFlow Hub link |
| MobileNet V1 0.50 128 | tensorflow-ic-imagenet-mobilenet-v1-050-128-classification-4 |
TensorFlow Hub link |
| MobileNet V1 0.25 224 | tensorflow-ic-imagenet-mobilenet-v1-025-224-classification-4 |
TensorFlow Hub link |
| MobileNet V1 0.25 192 | tensorflow-ic-imagenet-mobilenet-v1-025-192-classification-4 |
TensorFlow Hub link |
| MobileNet V1 0.25 160 | tensorflow-ic-imagenet-mobilenet-v1-025-160-classification-4 |
TensorFlow Hub link |
| MobileNet V1 0.25 128 | tensorflow-ic-imagenet-mobilenet-v1-025-128-classification-4 |
TensorFlow Hub link |
| BiT-S R50x1 | tensorflow-ic-bit-s-r50x1-ilsvrc2012-classification-1 |
TensorFlow Hub link |
| BiT-S R50x3 | tensorflow-ic-bit-s-r50x3-ilsvrc2012-classification-1 |
TensorFlow Hub link |
| BiT-S R101x1 | tensorflow-ic-bit-s-r101x1-ilsvrc2012-classification-1 |
TensorFlow Hub link |
| BiT-S R101x3 | tensorflow-ic-bit-s-r101x3-ilsvrc2012-classification-1 |
TensorFlow Hub link |
| BiT-M R50x1 | tensorflow-ic-bit-m-r50x1-ilsvrc2012-classification-1 |
TensorFlow Hub link |
| BiT-M R50x3 | tensorflow-ic-bit-m-r50x3-ilsvrc2012-classification-1 |
TensorFlow Hub link |
| BiT-M R101x1 | tensorflow-ic-bit-m-r101x1-ilsvrc2012-classification-1 |
TensorFlow Hub link |
| BiT-M R101x3 | tensorflow-ic-bit-m-r101x3-ilsvrc2012-classification-1 |
TensorFlow Hub link |
| BiT-M R50x1 ImageNet-21k | tensorflow-ic-bit-m-r50x1-imagenet21k-classification-1 |
TensorFlow Hub link |
| BiT-M R50x3 ImageNet-21k | tensorflow-ic-bit-m-r50x3-imagenet21k-classification-1 |
TensorFlow Hub link |
| BiT-M R101x1 ImageNet-21k | tensorflow-ic-bit-m-r101x1-imagenet21k-classification-1 |
TensorFlow Hub link |
| BiT-M R101x3 ImageNet-21k | tensorflow-ic-bit-m-r101x3-imagenet21k-classification-1 |
TensorFlow Hub link |