kaira.models.image.Bourtsoulatze2019DeepJSCCEncoder

Inheritance diagram of Bourtsoulatze2019DeepJSCCEncoder

Inheritance diagram for Bourtsoulatze2019DeepJSCCEncoder

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

Bases: BaseModel

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

This model encodes the input image into a latent representation for transmission.

Parameters:

num_transmitted_filters – Number of filters in the final encoding layer

Methods

__init__

Initialize the Bourtsoulatze2019DeepJSCCEncoder.

forward

Forward pass through the encoder.

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

Initialize the Bourtsoulatze2019DeepJSCCEncoder.

Parameters:
  • num_transmitted_filters – Number of filters in the final encoding layer.

  • *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 encoder.

Parameters:
  • x – Input image tensor of shape (B, 3, H, W)

  • *args – Additional positional arguments (unused).

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

Returns:

Encoded representation of shape (B, num_transmitted_filters, H//4, W//4)