Companion

object Companion
Content copied to clipboard

Functions

fromOutput
Link copied to clipboard
@JvmStatic
fun fromOutput(finalLayer: Layer): Functional
Content copied to clipboard

Creates the Functional model.

loadDefaultModelConfiguration
Link copied to clipboard
@JvmStatic
fun loadDefaultModelConfiguration(modelDirectory: File, inputShape: IntArray? = null): Functional
Content copied to clipboard

Loads a Functional model from json file with name 'modelConfig.json' with model configuration located in modelDirectory.

loadModelConfiguration
Link copied to clipboard
@JvmStatic
fun loadModelConfiguration(configuration: File, inputShape: IntArray? = null): Functional
Content copied to clipboard

Loads a Functional model from json file with model configuration.

loadModelLayersFromConfiguration
Link copied to clipboard
@JvmStatic
fun loadModelLayersFromConfiguration(configuration: File, inputShape: IntArray? = null): MutableList<Layer>
Content copied to clipboard

Loads a Functional model layers from json file with model configuration.

loadModelLayersFromDefaultConfiguration
Link copied to clipboard
@JvmStatic
fun loadModelLayersFromDefaultConfiguration(modelDirectory: File, inputShape: IntArray? = null): MutableList<Layer>
Content copied to clipboard

Loads a Functional model layers from json file with name 'modelConfig.json' with model configuration located in modelDirectory.

of
Link copied to clipboard
@JvmStatic
fun of(vararg layers: Layer, noInput: Boolean = false): Functional
Content copied to clipboard
@JvmStatic
fun of(layers: List<Layer>, noInput: Boolean = false): Functional
Content copied to clipboard

Creates the Functional model.

@JvmStatic
fun of(pretrainedModel: GraphTrainableModel, topModel: GraphTrainableModel): Functional
Content copied to clipboard

Creates the Functional model from two models: pretrainedModel and topModel. All layers of pretrainedModel will be frozen automatically. The input of the topModel will be connected to the output of the pretrainedModel.