org.deeplearning4j.nn.modelimport.keras.layers.convolutional

Class KerasConvolutionUtils

java.lang.Object

org.deeplearning4j.nn.modelimport.keras.layers.convolutional.KerasConvolutionUtils

public class KerasConvolutionUtils
extends Object

Constructor Summary

Constructors

Constructor and Description
KerasConvolutionUtils()

Method Summary

Modifier and Type	Method and Description
static ConvolutionMode	getConvolutionModeFromConfig(Map<String,Object> layerConfig, KerasLayerConfiguration conf) Get convolution border mode from Keras layer configuration.
static CNN2DFormat	getDataFormatFromConfig(Map<String,Object> layerConfig, KerasLayerConfiguration layerConfiguration) Return the CNN2DFormat from the configuration .
static int[]	getDilationRate(Map<String,Object> layerConfig, int dimension, KerasLayerConfiguration conf, boolean forceDilation) Get atrous / dilation rate from config
static int[]	getKernelSizeFromConfig(Map<String,Object> layerConfig, int dimension, KerasLayerConfiguration conf, int kerasMajorVersion) Get (convolution) kernel size from Keras layer configuration.
static int[]	getPaddingFromBorderModeConfig(Map<String,Object> layerConfig, int dimension, KerasLayerConfiguration conf, int kerasMajorVersion) Get (convolution) padding from Keras layer configuration.
static int[]	getStrideFromConfig(Map<String,Object> layerConfig, int dimension, KerasLayerConfiguration conf) Get (convolution) stride from Keras layer configuration.

Methods inherited from class java.lang.Object

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

Constructor Detail

KerasConvolutionUtils

public KerasConvolutionUtils()

Method Detail

getStrideFromConfig

public static int[] getStrideFromConfig(Map<String,Object> layerConfig,
                                        int dimension,
                                        KerasLayerConfiguration conf)
                                 throws InvalidKerasConfigurationException

Get (convolution) stride from Keras layer configuration.

Parameters:

layerConfig - dictionary containing Keras layer configuration

Returns:

Strides array from Keras configuration

Throws:

InvalidKerasConfigurationException - Invalid Keras config

getDilationRate

public static int[] getDilationRate(Map<String,Object> layerConfig,
                                    int dimension,
                                    KerasLayerConfiguration conf,
                                    boolean forceDilation)
                             throws InvalidKerasConfigurationException

Get atrous / dilation rate from config

Parameters:

layerConfig - dictionary containing Keras layer configuration

dimension - dimension of the convolution layer (1 or 2)

conf - Keras layer configuration

forceDilation - boolean to indicate if dilation argument should be in config

Returns:

list of integers with atrous rates

Throws:

InvalidKerasConfigurationException - Invalid Keras config

getDataFormatFromConfig

public static CNN2DFormat getDataFormatFromConfig(Map<String,Object> layerConfig,
                                                  KerasLayerConfiguration layerConfiguration)
                                           throws InvalidKerasConfigurationException

Return the CNN2DFormat from the configuration . If the value is KerasLayerConfiguration#getDIM_ORDERING_TENSORFLOW() then the value is CNN2DFormat.NHWC else it's KerasLayerConfiguration#getDIM_ORDERING_THEANO() which is CNN2DFormat.NCHW

Parameters:

layerConfig - the layer configuration to get the values from

layerConfiguration - the keras configuration used for retrieving values from the configuration

Returns:

the CNN2DFormat given the configuration

Throws:

InvalidKerasConfigurationException

getKernelSizeFromConfig

public static int[] getKernelSizeFromConfig(Map<String,Object> layerConfig,
                                            int dimension,
                                            KerasLayerConfiguration conf,
                                            int kerasMajorVersion)
                                     throws InvalidKerasConfigurationException

Get (convolution) kernel size from Keras layer configuration.

Parameters:

layerConfig - dictionary containing Keras layer configuration

Returns:

Convolutional kernel sizes

Throws:

InvalidKerasConfigurationException - Invalid Keras config

getConvolutionModeFromConfig

public static ConvolutionMode getConvolutionModeFromConfig(Map<String,Object> layerConfig,
                                                           KerasLayerConfiguration conf)
                                                    throws InvalidKerasConfigurationException,
                                                           UnsupportedKerasConfigurationException

Get convolution border mode from Keras layer configuration.

Parameters:

layerConfig - dictionary containing Keras layer configuration

Returns:

Border mode of convolutional layers

Throws:

InvalidKerasConfigurationException - Invalid Keras configuration

UnsupportedKerasConfigurationException

getPaddingFromBorderModeConfig

public static int[] getPaddingFromBorderModeConfig(Map<String,Object> layerConfig,
                                                   int dimension,
                                                   KerasLayerConfiguration conf,
                                                   int kerasMajorVersion)
                                            throws InvalidKerasConfigurationException,
                                                   UnsupportedKerasConfigurationException

Get (convolution) padding from Keras layer configuration.

Parameters:

layerConfig - dictionary containing Keras layer configuration

Returns:

Padding values derived from border mode

Throws:

InvalidKerasConfigurationException - Invalid Keras config

UnsupportedKerasConfigurationException