tf.keras.applications.mobilenet.decode_predictions
Decodes the prediction of an ImageNet model.
tf.keras.applications.mobilenet.decode_predictions(
preds, top=5
)
| Arguments | |
|---|---|
preds |
Numpy array encoding a batch of predictions. |
top |
Integer, how many top-guesses to return. Defaults to 5. |
| Returns | |
|---|---|
A list of lists of top class prediction tuples (class_name, class_description, score). One list of tuples per sample in batch input. |
| Raises | |
|---|---|
ValueError |
In case of invalid shape of the pred array (must be 2D). |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/keras/applications/mobilenet/decode_predictions