cellmap_data.transforms.targets.distance

cellmap_data.transforms.targets.distance#

Functions

transform(x)

Classes

DistanceTransform([use_cuda])

Initialize the distance transform.

SignedDistanceTransform([use_cuda])

Initialize the signed distance transform.
cellmap_data.transforms.targets.distance.transform(x: Tensor) → Tensor[source]#
Parameters:

x (Tensor)

Return type:

Tensor

class cellmap_data.transforms.targets.distance.DistanceTransform(use_cuda: bool = False)[source]#

Initialize the distance transform.

Parameters:

use_cuda (bool, optional) – Use CUDA. Defaults to False.

Raises:

NotImplementedError – CUDA is not supported yet.

forward(x: Tensor) → Tensor[source]#

Forward pass.

Parameters:

x (Tensor)

Return type:

Tensor

class cellmap_data.transforms.targets.distance.SignedDistanceTransform(use_cuda: bool = False)[source]#

Initialize the signed distance transform.

Parameters:

use_cuda (bool, optional) – Use CUDA. Defaults to False.

Raises:

NotImplementedError – CUDA is not supported yet.

forward(x: Tensor) → Tensor[source]#

Forward pass.

Parameters:

x (Tensor)

Return type:

Tensor

Contents