cellmap_data.transforms.augment.nan_to_num

cellmap_data.transforms.augment.nan_to_num#

Classes

NaNtoNum(params)

Initialize the NaN to number transformation.
class cellmap_data.transforms.augment.nan_to_num.NaNtoNum(params: Dict[str, Any])[source]#

Initialize the NaN to number transformation.

Parameters:

params (Dict[str, Any]) – Parameters for the transformation. Defaults to {}, see https://pytorch.org/docs/stable/generated/torch.nan_to_num.html for details.

transform(x: Any, params: Dict[str, Any] | None = None) → Any[source]#

Transform the input.

Parameters:

  • x (Any)

  • params (Dict[str, Any] | None)

Return type:

Any