org.deeplearning4j.nn.modelimport.keras.trainedmodels

Enum TrainedModels

java.lang.Object

java.lang.Enum<TrainedModels>

org.deeplearning4j.nn.modelimport.keras.trainedmodels.TrainedModels

All Implemented Interfaces:

java.io.Serializable, java.lang.Comparable<TrainedModels>

public enum TrainedModels
extends java.lang.Enum<TrainedModels>

Support for popular trained image-classification models

Enum Constant Summary

Enum Constants

Enum Constant and Description
VGG16
VGG16NOTOP

Method Summary

Modifier and Type	Method and Description
java.lang.String	decodePredictions(org.nd4j.linalg.api.ndarray.INDArray predictions)
protected java.lang.String	getH5FileName()
protected java.lang.String	getH5URL()
int[]	getInputShape() Shape of the input to the net, for a minibatch size of 1
protected java.lang.String	getJSONFileName()
protected java.lang.String	getJSONURL()
protected java.lang.String	getModelDir() Name of the sub dir in the local cache associated with the model.
int[]	getOuputShape() Shape of the output NDArray from the net, for a minibatch size of 1
org.nd4j.linalg.dataset.api.DataSetPreProcessor	getPreProcessor()
static TrainedModels	valueOf(java.lang.String name) Returns the enum constant of this type with the specified name.
static TrainedModels[]	values() Returns an array containing the constants of this enum type, in the order they are declared.

Methods inherited from class java.lang.Enum

clone, compareTo, equals, finalize, getDeclaringClass, hashCode, name, ordinal, toString, valueOf

Methods inherited from class java.lang.Object

getClass, notify, notifyAll, wait, wait, wait

Enum Constant Detail

VGG16

public static final TrainedModels VGG16

VGG16NOTOP

public static final TrainedModels VGG16NOTOP

Method Detail

values

public static TrainedModels[] values()

Returns an array containing the constants of this enum type, in the order they are declared. This method may be used to iterate over the constants as follows:

for (TrainedModels c : TrainedModels.values())
    System.out.println(c);

Returns:

an array containing the constants of this enum type, in the order they are declared

valueOf

public static TrainedModels valueOf(java.lang.String name)

Returns the enum constant of this type with the specified name. The string must match exactly an identifier used to declare an enum constant in this type. (Extraneous whitespace characters are not permitted.)

Parameters:

name - the name of the enum constant to be returned.

Returns:

the enum constant with the specified name

Throws:

java.lang.IllegalArgumentException - if this enum type has no constant with the specified name

java.lang.NullPointerException - if the argument is null

getModelDir

protected java.lang.String getModelDir()

Name of the sub dir in the local cache associated with the model. Local cache dir ~/.dl4j/trainedmodels

getJSONURL

protected java.lang.String getJSONURL()

getH5URL

protected java.lang.String getH5URL()

getH5FileName

protected java.lang.String getH5FileName()

getJSONFileName

protected java.lang.String getJSONFileName()

getPreProcessor

public org.nd4j.linalg.dataset.api.DataSetPreProcessor getPreProcessor()

Returns:

DataSetPreProcessor required for use with the model

getInputShape

public int[] getInputShape()

Shape of the input to the net, for a minibatch size of 1

Returns:

getOuputShape

public int[] getOuputShape()

Shape of the output NDArray from the net, for a minibatch size of 1

Returns:

decodePredictions

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