org.deeplearning4j.nn.modelimport.keras

Class KerasModelImport

java.lang.Object

org.deeplearning4j.nn.modelimport.keras.KerasModelImport

public class KerasModelImport
extends Object

Reads stored Keras configurations and weights from one of two archives: either (1) a single HDF5 file storing model and training JSON configurations and weights or (2) separate text file storing model JSON configuration and HDF5 file storing weights.

Author:

dave@skymind.io

Constructor Summary

Constructors

Constructor and Description
KerasModelImport()

Method Summary

Modifier and Type	Method and Description
static ComputationGraph	importKerasModelAndWeights(InputStream modelHdf5Stream) Load Keras (Functional API) Model saved using model.save_model(...).
static ComputationGraph	importKerasModelAndWeights(InputStream modelHdf5Stream, boolean enforceTrainingConfig) Load Keras (Functional API) Model saved using model.save_model(...).
static ComputationGraph	importKerasModelAndWeights(String modelHdf5Filename) Load Keras (Functional API) Model saved using model.save_model(...).
static ComputationGraph	importKerasModelAndWeights(String modelHdf5Filename, boolean enforceTrainingConfig) Load Keras (Functional API) Model saved using model.save_model(...).
static ComputationGraph	importKerasModelAndWeights(String modelJsonFilename, String weightsHdf5Filename) Load Keras (Functional API) Model for which the configuration and weights were saved separately using calls to model.to_json() and model.save_weights(...).
static ComputationGraph	importKerasModelAndWeights(String modelJsonFilename, String weightsHdf5Filename, boolean enforceTrainingConfig) Load Keras (Functional API) Model for which the configuration and weights were saved separately using calls to model.to_json() and model.save_weights(...).
static ComputationGraphConfiguration	importKerasModelConfiguration(String modelJsonFilename) Load Keras (Functional API) Model for which the configuration was saved separately using calls to model.to_json() and model.save_weights(...).
static ComputationGraphConfiguration	importKerasModelConfiguration(String modelJsonFilename, boolean enforceTrainingConfig) Load Keras (Functional API) Model for which the configuration was saved separately using calls to model.to_json() and model.save_weights(...).
static MultiLayerConfiguration	importKerasSequentialConfiguration(String modelJsonFilename) Load Keras Sequential model for which the configuration was saved separately using calls to model.to_json() and model.save_weights(...).
static MultiLayerConfiguration	importKerasSequentialConfiguration(String modelJsonFilename, boolean enforceTrainingConfig) Load Keras Sequential model for which the configuration was saved separately using calls to model.to_json() and model.save_weights(...).
static MultiLayerNetwork	importKerasSequentialModelAndWeights(InputStream modelHdf5Stream) Load Keras Sequential model saved using model.save_model(...).
static MultiLayerNetwork	importKerasSequentialModelAndWeights(InputStream modelHdf5Stream, boolean enforceTrainingConfig) Load Keras Sequential model saved using model.save_model(...).
static MultiLayerNetwork	importKerasSequentialModelAndWeights(String modelHdf5Filename) Load Keras Sequential model saved using model.save_model(...).
static MultiLayerNetwork	importKerasSequentialModelAndWeights(String modelHdf5Filename, boolean enforceTrainingConfig) Load Keras Sequential model saved using model.save_model(...).
static MultiLayerNetwork	importKerasSequentialModelAndWeights(String modelJsonFilename, String weightsHdf5Filename) Load Keras Sequential model for which the configuration and weights were saved separately using calls to model.to_json() and model.save_weights(...).
static MultiLayerNetwork	importKerasSequentialModelAndWeights(String modelJsonFilename, String weightsHdf5Filename, boolean enforceTrainingConfig) Load Keras Sequential model for which the configuration and weights were saved separately using calls to model.to_json() and model.save_weights(...).

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

KerasModelImport

public KerasModelImport()

Method Detail

importKerasModelAndWeights

public static ComputationGraph importKerasModelAndWeights(InputStream modelHdf5Stream,
                                                          boolean enforceTrainingConfig)

Load Keras (Functional API) Model saved using model.save_model(...).

Parameters:

modelHdf5Stream - InputStream containing HDF5 archive storing Keras Model

enforceTrainingConfig - whether to enforce training configuration options

Returns:

ComputationGraph

See Also:

ComputationGraph

importKerasModelAndWeights

public static ComputationGraph importKerasModelAndWeights(InputStream modelHdf5Stream)

Load Keras (Functional API) Model saved using model.save_model(...).

Parameters:

modelHdf5Stream - InputStream containing HDF5 archive storing Keras Model

Returns:

ComputationGraph

See Also:

ComputationGraph

importKerasSequentialModelAndWeights

public static MultiLayerNetwork importKerasSequentialModelAndWeights(InputStream modelHdf5Stream,
                                                                     boolean enforceTrainingConfig)
                                                              throws IOException,
                                                                     InvalidKerasConfigurationException,
                                                                     UnsupportedKerasConfigurationException

Load Keras Sequential model saved using model.save_model(...).

Parameters:

modelHdf5Stream - InputStream containing HDF5 archive storing Keras Sequential model

enforceTrainingConfig - whether to enforce training configuration options

Returns:

ComputationGraph

Throws:

IOException

InvalidKerasConfigurationException

UnsupportedKerasConfigurationException

See Also:

ComputationGraph

importKerasSequentialModelAndWeights

public static MultiLayerNetwork importKerasSequentialModelAndWeights(InputStream modelHdf5Stream)
                                                              throws IOException,
                                                                     InvalidKerasConfigurationException,
                                                                     UnsupportedKerasConfigurationException

Load Keras Sequential model saved using model.save_model(...).

Parameters:

modelHdf5Stream - InputStream containing HDF5 archive storing Keras Sequential model

Returns:

ComputationGraph

Throws:

IOException

InvalidKerasConfigurationException

UnsupportedKerasConfigurationException

See Also:

ComputationGraph

importKerasModelAndWeights

public static ComputationGraph importKerasModelAndWeights(String modelHdf5Filename,
                                                          boolean enforceTrainingConfig)
                                                   throws IOException,
                                                          UnsupportedKerasConfigurationException,
                                                          InvalidKerasConfigurationException

Load Keras (Functional API) Model saved using model.save_model(...).

Parameters:

modelHdf5Filename - path to HDF5 archive storing Keras Model

enforceTrainingConfig - whether to enforce training configuration options

Returns:

ComputationGraph

Throws:

IOException

InvalidKerasConfigurationException

UnsupportedKerasConfigurationException

See Also:

ComputationGraph

importKerasModelAndWeights

public static ComputationGraph importKerasModelAndWeights(String modelHdf5Filename)
                                                   throws IOException,
                                                          UnsupportedKerasConfigurationException,
                                                          InvalidKerasConfigurationException

Load Keras (Functional API) Model saved using model.save_model(...).

Parameters:

modelHdf5Filename - path to HDF5 archive storing Keras Model

Returns:

ComputationGraph

Throws:

IOException

InvalidKerasConfigurationException

UnsupportedKerasConfigurationException

See Also:

ComputationGraph

importKerasSequentialModelAndWeights

public static MultiLayerNetwork importKerasSequentialModelAndWeights(String modelHdf5Filename,
                                                                     boolean enforceTrainingConfig)
                                                              throws IOException,
                                                                     InvalidKerasConfigurationException,
                                                                     UnsupportedKerasConfigurationException

Load Keras Sequential model saved using model.save_model(...).

Parameters:

modelHdf5Filename - path to HDF5 archive storing Keras Sequential model

enforceTrainingConfig - whether to enforce training configuration options

Returns:

MultiLayerNetwork

Throws:

IOException

InvalidKerasConfigurationException

UnsupportedKerasConfigurationException

See Also:

MultiLayerNetwork

importKerasSequentialModelAndWeights

public static MultiLayerNetwork importKerasSequentialModelAndWeights(String modelHdf5Filename)
                                                              throws IOException,
                                                                     InvalidKerasConfigurationException,
                                                                     UnsupportedKerasConfigurationException

Load Keras Sequential model saved using model.save_model(...).

Parameters:

modelHdf5Filename - path to HDF5 archive storing Keras Sequential model

Returns:

MultiLayerNetwork

Throws:

IOException

InvalidKerasConfigurationException

UnsupportedKerasConfigurationException

See Also:

MultiLayerNetwork

importKerasModelAndWeights

public static ComputationGraph importKerasModelAndWeights(String modelJsonFilename,
                                                          String weightsHdf5Filename,
                                                          boolean enforceTrainingConfig)
                                                   throws IOException,
                                                          InvalidKerasConfigurationException,
                                                          UnsupportedKerasConfigurationException

Load Keras (Functional API) Model for which the configuration and weights were saved separately using calls to model.to_json() and model.save_weights(...).

Parameters:

modelJsonFilename - path to JSON file storing Keras Model configuration

weightsHdf5Filename - path to HDF5 archive storing Keras model weights

enforceTrainingConfig - whether to enforce training configuration options

Returns:

ComputationGraph

Throws:

IOException

InvalidKerasConfigurationException

UnsupportedKerasConfigurationException

See Also:

ComputationGraph

importKerasModelAndWeights

public static ComputationGraph importKerasModelAndWeights(String modelJsonFilename,
                                                          String weightsHdf5Filename)
                                                   throws IOException,
                                                          InvalidKerasConfigurationException,
                                                          UnsupportedKerasConfigurationException

Load Keras (Functional API) Model for which the configuration and weights were saved separately using calls to model.to_json() and model.save_weights(...).

Parameters:

modelJsonFilename - path to JSON file storing Keras Model configuration

weightsHdf5Filename - path to HDF5 archive storing Keras model weights

Returns:

ComputationGraph

Throws:

IOException

InvalidKerasConfigurationException

UnsupportedKerasConfigurationException

See Also:

ComputationGraph

importKerasSequentialModelAndWeights

public static MultiLayerNetwork importKerasSequentialModelAndWeights(String modelJsonFilename,
                                                                     String weightsHdf5Filename,
                                                                     boolean enforceTrainingConfig)
                                                              throws IOException,
                                                                     InvalidKerasConfigurationException,
                                                                     UnsupportedKerasConfigurationException

Load Keras Sequential model for which the configuration and weights were saved separately using calls to model.to_json() and model.save_weights(...).

Parameters:

modelJsonFilename - path to JSON file storing Keras Sequential model configuration

weightsHdf5Filename - path to HDF5 archive storing Keras model weights

enforceTrainingConfig - whether to enforce training configuration options

Returns:

MultiLayerNetwork

Throws:

IOException

InvalidKerasConfigurationException

UnsupportedKerasConfigurationException

See Also:

MultiLayerNetwork

importKerasSequentialModelAndWeights

public static MultiLayerNetwork importKerasSequentialModelAndWeights(String modelJsonFilename,
                                                                     String weightsHdf5Filename)
                                                              throws IOException,
                                                                     InvalidKerasConfigurationException,
                                                                     UnsupportedKerasConfigurationException

Load Keras Sequential model for which the configuration and weights were saved separately using calls to model.to_json() and model.save_weights(...).

Parameters:

modelJsonFilename - path to JSON file storing Keras Sequential model configuration

weightsHdf5Filename - path to HDF5 archive storing Keras model weights

Returns:

MultiLayerNetwork

Throws:

IOException

InvalidKerasConfigurationException

UnsupportedKerasConfigurationException

See Also:

MultiLayerNetwork

importKerasModelConfiguration

public static ComputationGraphConfiguration importKerasModelConfiguration(String modelJsonFilename,
                                                                          boolean enforceTrainingConfig)
                                                                   throws IOException,
                                                                          InvalidKerasConfigurationException,
                                                                          UnsupportedKerasConfigurationException

Load Keras (Functional API) Model for which the configuration was saved separately using calls to model.to_json() and model.save_weights(...).

Parameters:

modelJsonFilename - path to JSON file storing Keras Model configuration

enforceTrainingConfig - whether to enforce training configuration options

Returns:

ComputationGraph

Throws:

IOException

InvalidKerasConfigurationException

UnsupportedKerasConfigurationException

See Also:

ComputationGraph

importKerasModelConfiguration

public static ComputationGraphConfiguration importKerasModelConfiguration(String modelJsonFilename)
                                                                   throws IOException,
                                                                          InvalidKerasConfigurationException,
                                                                          UnsupportedKerasConfigurationException

Load Keras (Functional API) Model for which the configuration was saved separately using calls to model.to_json() and model.save_weights(...).

Parameters:

modelJsonFilename - path to JSON file storing Keras Model configuration

Returns:

ComputationGraph

Throws:

IOException

InvalidKerasConfigurationException

UnsupportedKerasConfigurationException

See Also:

ComputationGraph

importKerasSequentialConfiguration

public static MultiLayerConfiguration importKerasSequentialConfiguration(String modelJsonFilename,
                                                                         boolean enforceTrainingConfig)
                                                                  throws IOException,
                                                                         InvalidKerasConfigurationException,
                                                                         UnsupportedKerasConfigurationException

Load Keras Sequential model for which the configuration was saved separately using calls to model.to_json() and model.save_weights(...).

Parameters:

modelJsonFilename - path to JSON file storing Keras Sequential model configuration

enforceTrainingConfig - whether to enforce training configuration options

Returns:

MultiLayerNetwork

Throws:

IOException

InvalidKerasConfigurationException

UnsupportedKerasConfigurationException

See Also:

MultiLayerNetwork

importKerasSequentialConfiguration

public static MultiLayerConfiguration importKerasSequentialConfiguration(String modelJsonFilename)
                                                                  throws IOException,
                                                                         InvalidKerasConfigurationException,
                                                                         UnsupportedKerasConfigurationException

Load Keras Sequential model for which the configuration was saved separately using calls to model.to_json() and model.save_weights(...).

Parameters:

modelJsonFilename - path to JSON file storing Keras Sequential model configuration

Returns:

MultiLayerNetwork

Throws:

IOException

InvalidKerasConfigurationException

UnsupportedKerasConfigurationException

See Also:

MultiLayerNetwork