ModelType

interface ModelType<T : InferenceModel, U : InferenceModel>

Basic interface for models loaded from S3.

Functions

model
Link copied to clipboard
open fun model(modelHub: ModelHub): T

Loads the model, identified by this name, from the modelHub.

preprocessInput
Link copied to clipboard
open fun preprocessInput(preprocessing: Preprocessing): FloatArray
abstract fun preprocessInput(data: FloatArray, tensorShape: LongArray): FloatArray

Common preprocessing function for the Neural Networks trained on ImageNet and whose weights are available with the keras.application.

pretrainedModel
Link copied to clipboard
abstract fun pretrainedModel(modelHub: ModelHub): U

Returns the specially prepared pre-trained model of the type U.

Properties

modelRelativePath
Link copied to clipboard
abstract val modelRelativePath: String

Relative path to model for local and S3 buckets storages.

Inheritors

TFModels
Link copied to clipboard