Shortcuts

SimMIMLinearDecoder

class mmpretrain.models.necks.SimMIMLinearDecoder(in_channels, encoder_stride)[源代码]

Linear Decoder For SimMIM pretraining.

This neck reconstructs the original image from the shrunk feature map.

参数:
  • in_channels (int) – Channel dimension of the feature map.

  • encoder_stride (int) – The total stride of the encoder.

forward(x)[源代码]

Forward function.