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 ModelinitPretrained()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.StringmodelName()booleanpretrainedAvailable(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:
modelNamein interfaceInstantiableModel
-
-