| View source on GitHub |
Decodes the prediction of an ImageNet model.
tf.keras.applications.resnet.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.3/api_docs/python/tf/keras/applications/resnet/decode_predictions