org.deeplearning4j.zoo.util.imagenet

Class ImageNetLabels

java.lang.Object

org.deeplearning4j.zoo.util.imagenet.ImageNetLabels

public class ImageNetLabels
extends Object

Helper class with a static method that returns the label description.

Author:

susaneraly

Constructor Summary

Constructors

Constructor and Description
ImageNetLabels()

Method Summary

Modifier and Type	Method and Description
String	decodePredictions(org.nd4j.linalg.api.ndarray.INDArray predictions) Given predictions from the trained model this method will return a string listing the top five matches and the respective probabilities
String	getLabel(int n) Returns the description of tne nth class in the 1000 classes of ImageNet.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

ImageNetLabels

public ImageNetLabels()

Method Detail

getLabel

public String getLabel(int n)

Returns the description of tne nth class in the 1000 classes of ImageNet.

Parameters:

n -

Returns:

decodePredictions

public String decodePredictions(org.nd4j.linalg.api.ndarray.INDArray predictions)

Given predictions from the trained model this method will return a string listing the top five matches and the respective probabilities

Parameters:

predictions -

Returns: