org.deeplearning4j.zoo.util

Interface Labels

All Known Implementing Classes:

BaseLabels, COCOLabels, DarknetLabels, ImageNetLabels, VOCLabels

public interface Labels

Interface to helper classes that return label descriptions.

Author:

saudet

Method Summary

Modifier and Type	Method and Description
List<List<ClassPrediction>>	decodePredictions(INDArray predictions, int n) Given predictions from the trained model this method will return a list of the top n matches and the respective probabilities.
String	getLabel(int n) Returns the description of the nth class from the classes of a dataset.

Method Detail

getLabel

String getLabel(int n)

Returns the description of the nth class from the classes of a dataset.

Parameters:

n -

Returns:

label description

decodePredictions

List<List<ClassPrediction>> decodePredictions(INDArray predictions,
                                              int n)

Given predictions from the trained model this method will return a list of the top n matches and the respective probabilities.

Parameters:

predictions - raw

Returns:

decoded predictions