org.deeplearning4j.nn.conf.layers

Class DepthwiseConvolution2D

java.lang.Object

org.deeplearning4j.nn.conf.layers.Layer

org.deeplearning4j.nn.conf.layers.BaseLayer

org.deeplearning4j.nn.conf.layers.FeedForwardLayer

org.deeplearning4j.nn.conf.layers.ConvolutionLayer

org.deeplearning4j.nn.conf.layers.DepthwiseConvolution2D

All Implemented Interfaces:

Serializable, Cloneable, TrainingConfig

public class DepthwiseConvolution2D
extends ConvolutionLayer

2D depth-wise convolution layer configuration.

Performs a channels-wise convolution, which operates on each of the input maps separately. A channel multiplier is used to specify the number of outputs per input map. This convolution is carried out with the specified kernel sizes, stride and padding values.

Author:

Max Pumperla

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	DepthwiseConvolution2D.Builder

Nested classes/interfaces inherited from class org.deeplearning4j.nn.conf.layers.ConvolutionLayer

ConvolutionLayer.AlgoMode, ConvolutionLayer.BaseConvBuilder<T extends ConvolutionLayer.BaseConvBuilder<T>>, ConvolutionLayer.BwdDataAlgo, ConvolutionLayer.BwdFilterAlgo, ConvolutionLayer.FwdAlgo

Field Summary

Fields

Modifier and Type	Field and Description
protected int	depthMultiplier

Fields inherited from class org.deeplearning4j.nn.conf.layers.ConvolutionLayer

cnn2dDataFormat, convolutionMode, cudnnAlgoMode, cudnnAllowFallback, cudnnBwdDataAlgo, cudnnBwdFilterAlgo, cudnnFwdAlgo, dilation, hasBias, kernelSize, padding, stride

Fields inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer

nIn, nOut, timeDistributedFormat

Fields inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer

activationFn, biasInit, biasUpdater, gainInit, gradientNormalization, gradientNormalizationThreshold, iUpdater, regularization, regularizationBias, weightInitFn, weightNoise

Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer

constraints, iDropout, layerName

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	DepthwiseConvolution2D(DepthwiseConvolution2D.Builder builder)

Method Summary

Modifier and Type	Method and Description
DepthwiseConvolution2D	clone()
InputType	getOutputType(int layerIndex, InputType inputType) For a given type of input to this layer, what is the type of the output?
ParamInitializer	initializer()
Layer	instantiate(NeuralNetConfiguration conf, Collection<TrainingListener> trainingListeners, int layerIndex, INDArray layerParamsView, boolean initializeParams, DataType networkDataType)
void	setNIn(InputType inputType, boolean override) Set the nIn value (number of inputs, or input channels for CNNs) based on the given input type

Methods inherited from class org.deeplearning4j.nn.conf.layers.ConvolutionLayer

getMemoryReport, getPreProcessorForInputType, hasBias

Methods inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer

isPretrainParam

Methods inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer

getGradientNormalization, getRegularizationByParam, getUpdaterByParam, resetLayerDefaultConfig

Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer

initializeConstraints, setDataType

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.deeplearning4j.nn.api.TrainingConfig

getGradientNormalizationThreshold, getLayerName

Field Detail

depthMultiplier

protected int depthMultiplier

Constructor Detail

DepthwiseConvolution2D

protected DepthwiseConvolution2D(DepthwiseConvolution2D.Builder builder)

Method Detail

clone

public DepthwiseConvolution2D clone()

Overrides:

clone in class ConvolutionLayer

instantiate

public Layer instantiate(NeuralNetConfiguration conf,
                         Collection<TrainingListener> trainingListeners,
                         int layerIndex,
                         INDArray layerParamsView,
                         boolean initializeParams,
                         DataType networkDataType)

Overrides:

instantiate in class ConvolutionLayer

initializer

public ParamInitializer initializer()

Overrides:

initializer in class ConvolutionLayer

Returns:

The parameter initializer for this model

getOutputType

public InputType getOutputType(int layerIndex,
                               InputType inputType)

Description copied from class: Layer

For a given type of input to this layer, what is the type of the output?

Overrides:

getOutputType in class ConvolutionLayer

Parameters:

layerIndex - Index of the layer

inputType - Type of input for the layer

Returns:

Type of output from the layer

setNIn

public void setNIn(InputType inputType,
                   boolean override)

Description copied from class: Layer

Set the nIn value (number of inputs, or input channels for CNNs) based on the given input type

Overrides:

setNIn in class ConvolutionLayer

Parameters:

inputType - Input type for this layer

override - If false: only set the nIn value if it's not already set. If true: set it regardless of whether it's already set or not.