Package smile.vision.layer

Class SqueezeExcitation

java.lang.Object

smile.deep.layer.LayerBlock

smile.vision.layer.SqueezeExcitation

All Implemented Interfaces:

Function<Tensor,Tensor>, Layer

public class SqueezeExcitation
extends LayerBlock

Squeeze-and-Excitation block from "Squeeze-and-Excitation Networks".

Field Summary

Fields inherited from class smile.deep.layer.LayerBlock

device, dtype, module

Constructor Summary

Constructors

Constructor	Description
SqueezeExcitation(int inputChannels, int squeezeChannels)	Constructor.
SqueezeExcitation(int inputChannels, int squeezeChannels, ActivationFunction delta, ActivationFunction sigma)	Constructor.

Method Summary

Modifier and Type	Method	Description
Tensor	forward(Tensor input)	Forward propagation (or forward pass) through the layer.

Methods inherited from class smile.deep.layer.LayerBlock

add, add, asTorch, device, dtype, eval, isTraining, load, save, to, to, toString, train

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.util.function.Function

andThen, compose

Methods inherited from interface smile.deep.layer.Layer

apply

Constructor Details

SqueezeExcitation

public SqueezeExcitation(int inputChannels,
 int squeezeChannels)

Constructor.

Parameters:

inputChannels - the number of channels in the input image.

squeezeChannels - the number of squeeze channels.

SqueezeExcitation

public SqueezeExcitation(int inputChannels,
 int squeezeChannels,
 ActivationFunction delta,
 ActivationFunction sigma)

Constructor.

Parameters:

inputChannels - the number of channels in the input image.

squeezeChannels - the number of squeeze channels.

delta - the delta activation function.

sigma - the sigma activation function.

Method Details

forward

public Tensor forward(Tensor input)

Description copied from interface: Layer

Forward propagation (or forward pass) through the layer.

Parameters:

input - the input tensor.

Returns:

the output tensor.