SqueezeExcitation

class torchvision.ops.SqueezeExcitation(input_channels: int, squeeze_channels: int, activation: ~typing.Callable[[...], ~torch.nn.modules.module.Module] = <class 'torch.nn.modules.activation.ReLU'>, scale_activation: ~typing.Callable[[...], ~torch.nn.modules.module.Module] = <class 'torch.nn.modules.activation.Sigmoid'>)

此模块实现了来自 https://arxiv.org/abs/1709.01507 的 Squeeze-and-Excitation 模块（见图 1）。参数 activation 和 scale_activation 对应于公式 3 中的 delta 和 sigma。

参数:

• input_channels (int) – 输入图像中的通道数

• squeeze_channels (int) – squeeze 通道数

• activation (Callable[..., torch.nn.Module], optional) – delta 激活函数。 默认值：torch.nn.ReLU

• scale_activation (Callable[..., torch.nn.Module]) – sigma 激活函数。 默认值：torch.nn.Sigmoid

forward(input: Tensor) → Tensor

定义每次调用时执行的计算。

应由所有子类重写。

注意

尽管 forward 过程的配方需要在该函数中定义，但应该在此之后调用 Module 实例而不是此函数，因为前者会处理已注册的钩子，而后者会静默地忽略它们。