weka.dl4j.zoo.keras

Class EfficientNet

java.lang.Object

org.deeplearning4j.zoo.ZooModel

weka.dl4j.zoo.keras.KerasZooModel

weka.dl4j.zoo.keras.EfficientNet

All Implemented Interfaces:

java.io.Serializable, org.deeplearning4j.zoo.InstantiableModel

public class EfficientNet
extends KerasZooModel

Wrapper class for the different versions of EfficientNet. Has the same interface as Dl4j zoo models, so we can simply call initPretrained().

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	EfficientNet.VARIATION Different variations of the model.

Field Summary

Fields

Modifier and Type	Field and Description
static int[]	inputShape Default input shape of the model.

Constructor Summary

Constructors

Constructor and Description
EfficientNet() Instantiate the model.

Method Summary

Modifier and Type	Method and Description
java.lang.String	modelFamily() Get the model family.
java.lang.String	modelPrettyName() Get the pretty name of the model (e.g., ResNet 50)
void	setVariation(java.lang.Enum variation)

Methods inherited from class weka.dl4j.zoo.keras.KerasZooModel

init, initPretrained, metaData, modelType, pretrainedChecksum, pretrainedUrl, setInputShape

Methods inherited from class org.deeplearning4j.zoo.ZooModel

initPretrained, modelName, pretrainedAvailable

Methods inherited from class java.lang.Object

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

Field Detail

inputShape

public static int[] inputShape

Default input shape of the model.

Constructor Detail

EfficientNet

public EfficientNet()

Instantiate the model.

Method Detail

setVariation

public void setVariation(java.lang.Enum variation)

Specified by:

setVariation in class KerasZooModel

modelFamily

public java.lang.String modelFamily()

Description copied from class: KerasZooModel

Get the model family.

Specified by:

modelFamily in class KerasZooModel

Returns:

String denoting the model family

modelPrettyName

public java.lang.String modelPrettyName()

Description copied from class: KerasZooModel

Get the pretty name of the model (e.g., ResNet 50)

Specified by:

modelPrettyName in class KerasZooModel

Returns:

Model name.