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

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.api.ndarray.INDArray	backprop(org.nd4j.linalg.api.ndarray.INDArray epsilons, int miniBatchSize)
TensorFlowCnnToFeedForwardPreProcessor	clone()
org.nd4j.linalg.api.ndarray.INDArray	preProcess(org.nd4j.linalg.api.ndarray.INDArray input, int miniBatchSize)

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(int inputHeight,
                                              int inputWidth,
                                              int numChannels)

TensorFlowCnnToFeedForwardPreProcessor

public TensorFlowCnnToFeedForwardPreProcessor(int inputHeight,
                                              int inputWidth)

TensorFlowCnnToFeedForwardPreProcessor

public TensorFlowCnnToFeedForwardPreProcessor()

Method Detail

preProcess

public org.nd4j.linalg.api.ndarray.INDArray preProcess(org.nd4j.linalg.api.ndarray.INDArray input,
                                                       int miniBatchSize)

Specified by:

preProcess in interface InputPreProcessor

Overrides:

preProcess in class CnnToFeedForwardPreProcessor

backprop

public org.nd4j.linalg.api.ndarray.INDArray backprop(org.nd4j.linalg.api.ndarray.INDArray epsilons,
                                                     int miniBatchSize)

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