initialize

fun initialize(width: Int, height: Int, modelShape: ModelShape)

Initialize and load the tensorflow model.