For image classification, two kinds of modules are available:
- Modules to do image classification with the particular classes that the module has been trained for.
- Modules to extract image feature vectors, (a.k.a. "bottleneck values") for use in custom image classifiers. (This is elaborated in the image retraining tutorial.)
Click on a module to view its documentation, or reference the URL from the TensorFlow Hub library like so:
m = hub.Module("https://tfhub.dev/...")
- Inception V1: classification, feature_vector.
- Inception V2: classification, feature_vector.
- Inception V3: classification, feature_vector.
- Inception-ResNet V2: classification, feature_vector.
MobileNets come in various sizes controlled by a multiplier for the depth (number of features), and trained for various sizes of input images. See the module documentation for details.
-
MobileNet V1
-
MobileNet V1 instrumented for quantization with TF-Lite ("/quantops")
-
MobileNet V2
- NASNet-A large: classification, feature_vector.
- NASNet-A mobile: classification, feature_vector.
- PNASNet-5 large: classification, feature_vector.
-
ResNet V1
50 layers 101 layers 152 layers classification
feature_vectorclassification
feature_vectorclassification
feature_vector -
ResNet V2
50 layers 101 layers 152 layers classification
feature_vectorclassification
feature_vectorclassification
feature_vector