org.deeplearning4j.zoo.util.darknet

Class DarknetLabels

java.lang.Object

org.deeplearning4j.zoo.util.BaseLabels

org.deeplearning4j.zoo.util.darknet.DarknetLabels

All Implemented Interfaces:

Labels

public class DarknetLabels
extends BaseLabels

Helper class that returns label descriptions for Darknet models trained with ImageNet.

Author:

saudet

Field Summary

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

labels

Constructor Summary

Constructors

Constructor and Description
DarknetLabels() Calls this(true).
DarknetLabels(boolean shortnames)
DarknetLabels(boolean shortnames, int numClasses)
DarknetLabels(int numClasses)

Method Summary

Modifier and Type	Method and Description
protected URL	getURL()
protected String	resourceMD5()
protected String	resourceName()

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

decodePredictions, getLabel, getLabels, getLabels, getResourceFile

Methods inherited from class java.lang.Object

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

Constructor Detail

DarknetLabels

public DarknetLabels()
              throws IOException

Calls this(true). Defaults to 1000 clasess

Throws:

IOException

DarknetLabels

public DarknetLabels(int numClasses)
              throws IOException

Parameters:

numClasses - Number of classes (usually 1000 or 9000, depending on the model)

Throws:

IOException

DarknetLabels

public DarknetLabels(boolean shortnames)
              throws IOException

Parameters:

shortnames - if true, uses "imagenet.shortnames.list", otherwise "imagenet.labels.list".

Throws:

IOException

DarknetLabels

public DarknetLabels(boolean shortnames,
                     int numClasses)
              throws IOException

Parameters:

shortnames - if true, uses "imagenet.shortnames.list", otherwise "imagenet.labels.list".

numClasses - Number of classes (usually 1000 or 9000, depending on the model)

Throws:

IOException

Method Detail

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()