org.bytedeco.caffe

Class FloatBatchNormLayer

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.caffe.FloatLayer

org.bytedeco.caffe.FloatBatchNormLayer

All Implemented Interfaces:

AutoCloseable

@Name(value="caffe::BatchNormLayer<float>")
 @NoOffset
 @Properties(inherit=caffe.class)
public class FloatBatchNormLayer
extends FloatLayer

\brief Normalizes the input to have 0-mean and/or unit (1) variance across the batch. This layer computes Batch Normalization as described in [1]. For each channel in the data (i.e. axis 1), it subtracts the mean and divides by the variance, where both statistics are computed across both spatial dimensions and across the different examples in the batch. By default, during training time, the network is computing global mean/variance statistics via a running average, which is then used at test time to allow deterministic outputs for each input. You can manually toggle whether the network is accumulating or using the statistics via the use_global_stats option. For reference, these statistics are kept in the layer's three blobs: (0) mean, (1) variance, and (2) moving average factor. Note that the original paper also included a per-channel learned bias and scaling factor. To implement this in Caffe, define a ScaleLayer configured with bias_term: true after each BatchNormLayer to handle both the bias and scaling factor. [1] S. Ioffe and C. Szegedy, "Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift." arXiv preprint arXiv:1502.03167 (2015). TODO(dox): thorough documentation for Forward, Backward, and proto params.

Nested Class Summary

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter

Field Summary

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
FloatBatchNormLayer(LayerParameter param)
FloatBatchNormLayer(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
protected void	Backward_cpu(FloatBlobVector top, BoolVector propagate_down, FloatBlobVector bottom)
protected void	Backward_gpu(FloatBlobVector top, BoolVector propagate_down, FloatBlobVector bottom)
int	ExactNumBottomBlobs() \brief Returns the exact number of bottom blobs required by the layer, or -1 if no exact number is required.
int	ExactNumTopBlobs() \brief Returns the exact number of top blobs required by the layer, or -1 if no exact number is required.
protected void	Forward_cpu(FloatBlobVector bottom, FloatBlobVector top)
protected void	Forward_gpu(FloatBlobVector bottom, FloatBlobVector top)
void	LayerSetUp(FloatBlobVector bottom, FloatBlobVector top) \brief Does layer-specific setup: your layer should implement this function as well as Reshape.
void	Reshape(FloatBlobVector bottom, FloatBlobVector top) \brief Adjust the shapes of top blobs and internal buffers to accommodate the shapes of the bottom blobs.
BytePointer	type() \brief Returns the layer type.

Methods inherited from class org.bytedeco.caffe.FloatLayer

AllowForceBackward, AutoTopBlobs, Backward, blobs, CheckBlobCounts, EqualNumBottomTopBlobs, Forward, layer_param, loss, MaxBottomBlobs, MaxTopBlobs, MinBottomBlobs, MinTopBlobs, param_propagate_down, set_loss, set_param_propagate_down, SetUp, ToProto

Methods inherited from class org.bytedeco.javacpp.Pointer

address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero

Methods inherited from class java.lang.Object

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

Constructor Detail

FloatBatchNormLayer

public FloatBatchNormLayer(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

FloatBatchNormLayer

public FloatBatchNormLayer(@Const @ByRef
                           LayerParameter param)

Method Detail

LayerSetUp

@Virtual
public void LayerSetUp(@Const @ByRef
                                FloatBlobVector bottom,
                                @Const @ByRef
                                FloatBlobVector top)

Description copied from class: FloatLayer

\brief Does layer-specific setup: your layer should implement this function as well as Reshape.

Overrides:

LayerSetUp in class FloatLayer

Parameters:

bottom - the preshaped input blobs, whose data fields store the input data for this layer

top - the allocated but unshaped output blobs This method should do one-time layer specific setup. This includes reading and processing relevent parameters from the layer_param_. Setting up the shapes of top blobs and internal buffers should be done in Reshape, which will be called before the forward pass to adjust the top blob sizes.

Reshape

@Virtual
public void Reshape(@Const @ByRef
                             FloatBlobVector bottom,
                             @Const @ByRef
                             FloatBlobVector top)

Description copied from class: FloatLayer

\brief Adjust the shapes of top blobs and internal buffers to accommodate the shapes of the bottom blobs.

Overrides:

Reshape in class FloatLayer

Parameters:

bottom - the input blobs, with the requested input shapes

top - the top blobs, which should be reshaped as needed This method should reshape top blobs as needed according to the shapes of the bottom (input) blobs, as well as reshaping any internal buffers and making any other necessary adjustments so that the layer can accommodate the bottom blobs.

type

@Virtual
 @Const(value={false,false,true})
 @Cast(value="const char*")
public BytePointer type()

Description copied from class: FloatLayer

\brief Returns the layer type.

Overrides:

type in class FloatLayer

ExactNumBottomBlobs

@Virtual
 @Const(value={false,false,true})
public int ExactNumBottomBlobs()

Description copied from class: FloatLayer

\brief Returns the exact number of bottom blobs required by the layer, or -1 if no exact number is required. This method should be overridden to return a non-negative value if your layer expects some exact number of bottom blobs.

Overrides:

ExactNumBottomBlobs in class FloatLayer

ExactNumTopBlobs

@Virtual
 @Const(value={false,false,true})
public int ExactNumTopBlobs()

Description copied from class: FloatLayer

\brief Returns the exact number of top blobs required by the layer, or -1 if no exact number is required. This method should be overridden to return a non-negative value if your layer expects some exact number of top blobs.

Overrides:

ExactNumTopBlobs in class FloatLayer

Forward_cpu

@Virtual
protected void Forward_cpu(@Const @ByRef
                                    FloatBlobVector bottom,
                                    @Const @ByRef
                                    FloatBlobVector top)

Overrides:

Forward_cpu in class FloatLayer

Forward_gpu

@Virtual
protected void Forward_gpu(@Const @ByRef
                                    FloatBlobVector bottom,
                                    @Const @ByRef
                                    FloatBlobVector top)

Overrides:

Forward_gpu in class FloatLayer

Backward_cpu

@Virtual
protected void Backward_cpu(@Const @ByRef
                                     FloatBlobVector top,
                                     @Const @ByRef
                                     BoolVector propagate_down,
                                     @Const @ByRef
                                     FloatBlobVector bottom)

Overrides:

Backward_cpu in class FloatLayer

Backward_gpu

@Virtual
protected void Backward_gpu(@Const @ByRef
                                     FloatBlobVector top,
                                     @Const @ByRef
                                     BoolVector propagate_down,
                                     @Const @ByRef
                                     FloatBlobVector bottom)

Overrides:

Backward_gpu in class FloatLayer