org.deeplearning4j.zoo

Class ZooModel<T>

java.lang.Object

org.deeplearning4j.zoo.ZooModel<T>

All Implemented Interfaces:

InstantiableModel

Direct Known Subclasses:

AlexNet, Darknet19, FaceNetNN4Small2, InceptionResNetV1, LeNet, NASNet, ResNet50, SimpleCNN, SqueezeNet, TextGenerationLSTM, TinyYOLO, UNet, VGG16, VGG19, Xception, YOLO2

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)

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.
<M extends Model> M	initPretrained(PretrainedType pretrainedType) Returns a pretrained model for the given dataset, if available.
String	modelName()
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

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 <M extends Model> M initPretrained(PretrainedType pretrainedType)
                                   throws IOException

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

Parameters:

pretrainedType -

Returns:

Throws:

IOException

modelName

public String modelName()

Specified by:

modelName in interface InstantiableModel