public
invertedresidual
— function
invertedresidual(kernel_size, inplanes, hidden_planes, outplanes, activation = relu;
stride, reduction = nothing)
Create a basic inverted residual block for MobileNet variants (reference).
Arguments
kernel_size
: The kernel size of the convolutional layersinplanes
: The number of input feature mapshidden_planes
: The number of feature maps in the hidden layeroutplanes
: The number of output feature mapsactivation
: The activation function for the first two convolution layerstride
: The stride of the convolutional kernel, has to be either 1 or 2reduction
: The reduction factor for the number of hidden feature maps in a squeeze and excite layer (seesqueeze_excite
). Must be ≥ 1 ornothing
for no squeeze and excite layer.