View source on GitHub
|
Decodes the prediction of an ImageNet model.
tf.keras.applications.efficientnet_v2.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).
|