Build an Encoder

mmit.create_encoder(name, in_chans=3, out_indices=None, output_stride=None, **kwargs)

Build an encoder from a name and some keyword arguments.

Parameters:
  • name (str) – The name of the encoder.

  • in_chans (int) – The number of input channels.

  • out_indices (Optional[tuple]) – The indices of the feature maps to return.

  • output_stride (Optional[int]) – The output stride of the encoder.

  • kwargs – Keyword arguments for the encoder. Take a look at the specific encoder docs for more info!

Return type:

Module