org.deeplearning4j.zoo.util.imagenet

Class ImageNetLabels

java.lang.Object

org.deeplearning4j.zoo.util.BaseLabels

org.deeplearning4j.zoo.util.imagenet.ImageNetLabels

All Implemented Interfaces:

Labels

public class ImageNetLabels
extends BaseLabels

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

Author:

susaneraly

Field Summary

Fields inherited from class org.deeplearning4j.zoo.util.BaseLabels

labels

Constructor Summary

Constructors

Constructor and Description
ImageNetLabels()

Method Summary

Modifier and Type	Method and Description
String	decodePredictions(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.
protected ArrayList<String>	getLabels() Override to return labels when not calling BaseLabels.BaseLabels(String).
protected URL	getURL()
protected String	resourceMD5()
protected String	resourceName()

Methods inherited from class org.deeplearning4j.zoo.util.BaseLabels

decodePredictions, getLabels, getResourceFile

Methods inherited from class java.lang.Object

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

Constructor Detail

ImageNetLabels

public ImageNetLabels()
               throws IOException

Throws:

IOException

Method Detail

getLabels

protected ArrayList<String> getLabels()
                               throws IOException

Description copied from class: BaseLabels

Override to return labels when not calling BaseLabels.BaseLabels(String).

Overrides:

getLabels in class BaseLabels

Throws:

IOException

getLabel

public String getLabel(int n)

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

Specified by:

getLabel in interface Labels

Overrides:

getLabel in class BaseLabels

Parameters:

n -

Returns:

getURL

protected URL getURL()

Specified by:

getURL in class BaseLabels

Returns:

URL of the resource to download

resourceName

protected String resourceName()

Specified by:

resourceName in class BaseLabels

Returns:

Name of the resource (used for inferring local storage parent directory)

resourceMD5

protected String resourceMD5()

Specified by:

resourceMD5 in class BaseLabels

Returns:

MD5 of the resource at getURL()

decodePredictions

public String decodePredictions(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: