org.deeplearning4j.nn.modelimport.keras.preprocessors

Class TensorFlowCnnToFeedForwardPreProcessor

java.lang.Object

org.deeplearning4j.nn.conf.preprocessor.CnnToFeedForwardPreProcessor

org.deeplearning4j.nn.modelimport.keras.preprocessors.TensorFlowCnnToFeedForwardPreProcessor

All Implemented Interfaces:

Serializable, Cloneable, InputPreProcessor

public class TensorFlowCnnToFeedForwardPreProcessor
extends CnnToFeedForwardPreProcessor

Specialized CnnToFeedForwardInputPreProcessor for use with Convolutional layers imported from Keras using the TensorFlow backend.

Author:

dave@skymind.io

See Also:

Serialized Form

Field Summary

Fields inherited from class org.deeplearning4j.nn.conf.preprocessor.CnnToFeedForwardPreProcessor

inputHeight, inputWidth, numChannels

Constructor Summary

Constructors

Constructor and Description
TensorFlowCnnToFeedForwardPreProcessor()
TensorFlowCnnToFeedForwardPreProcessor(long inputHeight, long inputWidth)
TensorFlowCnnToFeedForwardPreProcessor(long inputHeight, long inputWidth, long numChannels)

Method Summary

Modifier and Type	Method and Description
INDArray	backprop(INDArray epsilons, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)
TensorFlowCnnToFeedForwardPreProcessor	clone()
INDArray	preProcess(INDArray input, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)

Methods inherited from class org.deeplearning4j.nn.conf.preprocessor.CnnToFeedForwardPreProcessor

canEqual, equals, feedForwardMaskArray, getInputHeight, getInputWidth, getNumChannels, getOutputType, hashCode, setInputHeight, setInputWidth, setNumChannels, toString

Methods inherited from class java.lang.Object

finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

TensorFlowCnnToFeedForwardPreProcessor

public TensorFlowCnnToFeedForwardPreProcessor(long inputHeight,
                                              long inputWidth,
                                              long numChannels)

TensorFlowCnnToFeedForwardPreProcessor

public TensorFlowCnnToFeedForwardPreProcessor(long inputHeight,
                                              long inputWidth)

TensorFlowCnnToFeedForwardPreProcessor

public TensorFlowCnnToFeedForwardPreProcessor()

Method Detail

preProcess

public INDArray preProcess(INDArray input,
                           int miniBatchSize,
                           LayerWorkspaceMgr workspaceMgr)

Specified by:

preProcess in interface InputPreProcessor

Overrides:

preProcess in class CnnToFeedForwardPreProcessor

backprop

public INDArray backprop(INDArray epsilons,
                         int miniBatchSize,
                         LayerWorkspaceMgr workspaceMgr)

Specified by:

backprop in interface InputPreProcessor

Overrides:

backprop in class CnnToFeedForwardPreProcessor

clone

public TensorFlowCnnToFeedForwardPreProcessor clone()

Specified by:

clone in interface InputPreProcessor

Overrides:

clone in class CnnToFeedForwardPreProcessor