org.deeplearning4j.nn.modelimport.keras.layers.pooling

Class KerasPoolingUtils

java.lang.Object

org.deeplearning4j.nn.modelimport.keras.layers.pooling.KerasPoolingUtils

public class KerasPoolingUtils
extends Object

Constructor Summary

Constructors

Constructor and Description
KerasPoolingUtils()

Method Summary

Modifier and Type	Method and Description
static int[]	mapGlobalPoolingDimensions(String className, KerasLayerConfiguration conf, KerasLayer.DimOrder dimOrder) Map Keras pooling layers to DL4J pooling dimensions.
static PoolingType	mapPoolingType(String className, KerasLayerConfiguration conf) Map Keras pooling layers to DL4J pooling types.

Methods inherited from class java.lang.Object

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

Constructor Detail

KerasPoolingUtils

public KerasPoolingUtils()

Method Detail

mapPoolingType

public static PoolingType mapPoolingType(String className,
                                         KerasLayerConfiguration conf)
                                  throws UnsupportedKerasConfigurationException

Map Keras pooling layers to DL4J pooling types.

Parameters:

className - name of the Keras pooling class

Returns:

DL4J pooling type

Throws:

UnsupportedKerasConfigurationException - Unsupported Keras config

mapGlobalPoolingDimensions

public static int[] mapGlobalPoolingDimensions(String className,
                                               KerasLayerConfiguration conf,
                                               KerasLayer.DimOrder dimOrder)
                                        throws UnsupportedKerasConfigurationException

Map Keras pooling layers to DL4J pooling dimensions.

Parameters:

className - name of the Keras pooling class

dimOrder - the dimension order to determine which pooling dimensions to use

Returns:

pooling dimensions as int array

Throws:

UnsupportedKerasConfigurationException - Unsupported Keras config