CARAFEPack¶
- class mmcv.ops.CARAFEPack(channels: int, scale_factor: int, up_kernel: int = 5, up_group: int = 1, encoder_kernel: int = 3, encoder_dilation: int = 1, compressed_channels: int = 64)[source]¶
A unified package of CARAFE upsampler that contains: 1) channel compressor 2) content encoder 3) CARAFE op.
Official implementation of ICCV 2019 paper CARAFE: Content-Aware ReAssembly of FEatures.
- Parameters
channels (int) – input feature channels
scale_factor (int) – upsample ratio
up_kernel (int) – kernel size of CARAFE op
up_group (int) – group size of CARAFE op
encoder_kernel (int) – kernel size of content encoder
encoder_dilation (int) – dilation of content encoder
compressed_channels (int) – output channels of channels compressor
- Returns
upsampled feature map
- forward(x: torch.Tensor) → torch.Tensor[source]¶
Defines the computation performed at every call.
Should be overridden by all subclasses.
Note
Although the recipe for forward pass needs to be defined within this function, one should call the
Module
instance afterwards instead of this since the former takes care of running the registered hooks while the latter silently ignores them.