Package 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
-
-
Constructor Summary
Constructors Constructor Description ZooModel()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description Model
initPretrained()
By default, will return a pretrained ImageNet if available.<M extends Model>
MinitPretrained(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
-
-
-
-
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 interfaceInstantiableModel
-
-