View source on GitHub |
Decodes the prediction of an ImageNet model.
tf.keras.applications.efficientnet.decode_predictions(
preds, top=5
)
Args | |
---|---|
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).
|