kaira.models.image.Bourtsoulatze2019DeepJSCCDecoder

Inheritance diagram of Bourtsoulatze2019DeepJSCCDecoder

Inheritance diagram for Bourtsoulatze2019DeepJSCCDecoder

class kaira.models.image.Bourtsoulatze2019DeepJSCCDecoder(num_transmitted_filters: int, *args: Any, **kwargs: Any)[source]

Bases: BaseModel

DeepJSCC decoder model from [Bourtsoulatze et al., 2019].

This model decodes the transmitted representation back into an image.

Parameters:

num_transmitted_filters – Number of filters in the transmitted representation

Methods

__init__

Initialize the Bourtsoulatze2019DeepJSCCDecoder.

forward

Forward pass through the decoder.

__init__(num_transmitted_filters: int, *args: Any, **kwargs: Any)[source]

Initialize the Bourtsoulatze2019DeepJSCCDecoder.

Parameters:
  • num_transmitted_filters – Number of filters in the transmitted representation.

  • *args – Variable positional arguments passed to the base class.

  • **kwargs – Variable keyword arguments passed to the base class.

forward(x: Tensor, *args: Any, **kwargs: Any) Tensor[source]

Forward pass through the decoder.

Parameters:
  • x – Encoded representation of shape (B, num_transmitted_filters, H//4, W//4)

  • *args – Additional positional arguments (unused).

  • **kwargs – Additional keyword arguments (unused).

Returns:

Decoded image tensor of shape (B, 3, H, W)