weka.dl4j.zoo.keras

Class KerasZooModel

java.lang.Object

org.deeplearning4j.zoo.ZooModel

weka.dl4j.zoo.keras.KerasZooModel

All Implemented Interfaces:

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

Direct Known Subclasses:

DenseNet, EfficientNet, InceptionResNetV2, InceptionV3, MobileNet, NASNet, ResNet, VGG, Xception

public abstract class KerasZooModel
extends org.deeplearning4j.zoo.ZooModel
implements java.io.Serializable

This class essentially copies DL4J's ZooModel class, allowing the custom Keras models to provide the same interface as those models provided by DL4J. It was originally created as KerasZooModels were loaded from the .h5 file, so they couldn't use DL4J's ZooModel implementation (as it loads model from the native .zip format). Loading from .h5 format is a little unreliable at times so this class uses the native DL4J .zip format. This implementation is different in that the pretrained URL and checksum are taken from the KerasConstants file - a global constant which defines these values. This is easier to manage than having URLs and checksums in each individual model class (as is done in DL4J).

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
KerasZooModel()

Method Summary

Modifier and Type	Method and Description
org.deeplearning4j.nn.graph.ComputationGraph	init()
org.deeplearning4j.nn.graph.ComputationGraph	initPretrained(org.deeplearning4j.zoo.PretrainedType pretrainedType)
org.deeplearning4j.zoo.ModelMetaData	metaData()
abstract java.lang.String	modelFamily() Get the model family.
abstract java.lang.String	modelPrettyName() Get the pretty name of the model (e.g., ResNet 50)
java.lang.Class<? extends org.deeplearning4j.nn.api.Model>	modelType()
long	pretrainedChecksum(org.deeplearning4j.zoo.PretrainedType pretrainedType)
java.lang.String	pretrainedUrl(org.deeplearning4j.zoo.PretrainedType pretrainedType)
void	setInputShape(int[][] inputShape)
abstract void	setVariation(java.lang.Enum variation)

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

Constructor Detail

KerasZooModel

public KerasZooModel()

Method Detail

modelFamily

public abstract java.lang.String modelFamily()

Get the model family.

Returns:

String denoting the model family

modelPrettyName

public abstract java.lang.String modelPrettyName()

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

Returns:

Model name.

setVariation

public abstract void setVariation(java.lang.Enum variation)

modelType

public java.lang.Class<? extends org.deeplearning4j.nn.api.Model> modelType()

Specified by:

modelType in interface org.deeplearning4j.zoo.InstantiableModel

init

public org.deeplearning4j.nn.graph.ComputationGraph init()

Specified by:

init in interface org.deeplearning4j.zoo.InstantiableModel

pretrainedUrl

public java.lang.String pretrainedUrl(org.deeplearning4j.zoo.PretrainedType pretrainedType)

Specified by:

pretrainedUrl in interface org.deeplearning4j.zoo.InstantiableModel

pretrainedChecksum

public long pretrainedChecksum(org.deeplearning4j.zoo.PretrainedType pretrainedType)

Specified by:

pretrainedChecksum in interface org.deeplearning4j.zoo.InstantiableModel

initPretrained

public org.deeplearning4j.nn.graph.ComputationGraph initPretrained(org.deeplearning4j.zoo.PretrainedType pretrainedType)
                                                            throws java.io.IOException

Overrides:

initPretrained in class org.deeplearning4j.zoo.ZooModel

Throws:

java.io.IOException

setInputShape

public void setInputShape(int[][] inputShape)

Specified by:

setInputShape in interface org.deeplearning4j.zoo.InstantiableModel

metaData

public org.deeplearning4j.zoo.ModelMetaData metaData()

Specified by:

metaData in interface org.deeplearning4j.zoo.InstantiableModel