ModelHub

abstract class ModelHub(cacheDirectory: File)

This is an abstract class which provides methods for loading models, its weights and labels (for prediction purposes) to the local cacheDirectory.

Since

0.2

Constructors

ModelHub

fun ModelHub(cacheDirectory: File)

Functions

get

operator fun <T : InferenceModel, U : InferenceModel> get(modelType: ModelType<T, U>): U

loadModel

abstract fun <T : InferenceModel, U : InferenceModel> loadModel(modelType: ModelType<T, U>, loadingMode: LoadingMode = LoadingMode.SKIP_LOADING_IF_EXISTS): T

Loads model configuration without weights.

loadPretrainedModel

fun <T : InferenceModel, U : InferenceModel> loadPretrainedModel(modelType: ModelType<T, U>, loadingMode: LoadingMode = LoadingMode.SKIP_LOADING_IF_EXISTS): U

Loads pretrained model of modelType from the ModelHub in loadingMode.

Properties

cacheDirectory

val cacheDirectory: File

The directory for all loaded models and datasets. It should be created before model loading and should have all required permissions for file writing/reading on your OS.

Inheritors

TFModelHub