org.deeplearning4j.zoo

Class ZooModel<T>

java.lang.Object

org.deeplearning4j.zoo.ZooModel<T>

All Implemented Interfaces:

InstantiableModel

Direct Known Subclasses:

AlexNet, FaceNetNN4Small2, GoogLeNet, InceptionResNetV1, LeNet, ResNet50, SimpleCNN, TextGenerationLSTM, VGG16, VGG19

public abstract class ZooModel<T>
extends Object
implements InstantiableModel

A zoo model is instantiable, returns information about itself, and can download pretrained models if available.

Author:

Justin Long (crockpotveggies)

Field Summary

Fields

Modifier and Type	Field and Description
static File	ROOT_CACHE_DIR

Constructor Summary

Constructors

Constructor and Description
ZooModel()

Method Summary

Modifier and Type	Method and Description
Model	initPretrained() By default, will return a pretrained ImageNet if available.
Model	initPretrained(PretrainedType pretrainedType) Returns a pretrained model for the given dataset, if available.
boolean	pretrainedAvailable(PretrainedType pretrainedType)

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.deeplearning4j.zoo.InstantiableModel

init, metaData, modelType, pretrainedChecksum, pretrainedUrl, setInputShape, zooType

Field Detail

ROOT_CACHE_DIR

public static File ROOT_CACHE_DIR

Constructor Detail

ZooModel

public ZooModel()

Method Detail

pretrainedAvailable

public boolean pretrainedAvailable(PretrainedType pretrainedType)

initPretrained

public Model initPretrained()
                     throws IOException

By default, will return a pretrained ImageNet if available.

Returns:

Throws:

IOException

initPretrained

public Model initPretrained(PretrainedType pretrainedType)
                     throws IOException

Returns a pretrained model for the given dataset, if available.

Parameters:

pretrainedType -

Returns:

Throws:

IOException