Change ImageSpec to ModelSpec.
PiperOrigin-RevId: 370850915
Change-Id: I1b125aca711f1126c6d357a22bcf3d713d9bf735
diff --git a/tensorflow/lite/g3doc/tutorials/model_maker_image_classification.ipynb b/tensorflow/lite/g3doc/tutorials/model_maker_image_classification.ipynb
index d31e899..f723c99 100644
--- a/tensorflow/lite/g3doc/tutorials/model_maker_image_classification.ipynb
+++ b/tensorflow/lite/g3doc/tutorials/model_maker_image_classification.ipynb
@@ -127,7 +127,7 @@
"from tflite_model_maker.config import ExportFormat\n",
"from tflite_model_maker.config import QuantizationConfig\n",
"from tflite_model_maker.image_classifier import DataLoader\n",
- "from tflite_model_maker.image_classifier import ImageSpec\n",
+ "from tflite_model_maker.image_classifier import ModelSpec\n",
"\n",
"import matplotlib.pyplot as plt"
]
@@ -776,7 +776,7 @@
"\n",
"Moreover, we could also switch to other new models that inputs an image and outputs a feature vector with TensorFlow Hub format.\n",
"\n",
- "As [Inception V3](https://tfhub.dev/google/imagenet/inception_v3/feature_vector/1) model as an example, we could define `inception_v3_spec` which is an object of `ImageSpec` and contains the specification of the Inception V3 model.\n",
+ "As [Inception V3](https://tfhub.dev/google/imagenet/inception_v3/feature_vector/1) model as an example, we could define `inception_v3_spec` which is an object of `ModelSpec` and contains the specification of the Inception V3 model.\n",
"\n",
"We need to specify the model name `name`, the url of the TensorFlow Hub model `uri`. Meanwhile, the default value of `input_image_shape` is `[224, 224]`. We need to change it to `[299, 299]` for Inception V3 model."
]
@@ -789,7 +789,7 @@
},
"outputs": [],
"source": [
- "inception_v3_spec = ImageSpec(\n",
+ "inception_v3_spec = ModelSpec(\n",
" uri='https://tfhub.dev/google/imagenet/inception_v3/feature_vector/1')\n",
"inception_v3_spec.input_image_shape = [299, 299]"
]
@@ -822,7 +822,7 @@
"source": [
"If we'd like to use the custom model that's not in TensorFlow Hub, we should create and export [ModelSpec](https://www.tensorflow.org/hub/api_docs/python/hub/ModuleSpec) in TensorFlow Hub.\n",
"\n",
- "Then start to define `ImageSpec` object like the process above."
+ "Then start to define `ModelSpec` object like the process above."
]
},
{
