org.deeplearning4j.zoo

Interface InstantiableModel

All Known Implementing Classes:

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

public interface InstantiableModel

Interface for defining a model that can be instantiated and return information about itself.

Method Summary

Modifier and Type	Method and Description
<M extends Model> M	init()
ModelMetaData	metaData() Deprecated. No longer used, will be removed in a future release
String	modelName()
Class<? extends Model>	modelType()
long	pretrainedChecksum(PretrainedType pretrainedType)
String	pretrainedUrl(PretrainedType pretrainedType)
void	setInputShape(int[][] inputShape)

Method Detail

setInputShape

void setInputShape(int[][] inputShape)

init

<M extends Model> M init()

metaData

@Deprecated
ModelMetaData metaData()

Deprecated. No longer used, will be removed in a future release

modelType

Class<? extends Model> modelType()

pretrainedUrl

String pretrainedUrl(PretrainedType pretrainedType)

pretrainedChecksum

long pretrainedChecksum(PretrainedType pretrainedType)

modelName

String modelName()