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

Parameters:

x (Tensor)

Return type:

Tensor

class cellmap_data.transforms.targets.distance.DistanceTransform(use_cuda: bool = False)

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

Forward pass.

Parameters:

x (Tensor)

Return type:

Tensor

class cellmap_data.transforms.targets.distance.SignedDistanceTransform(use_cuda: bool = False)

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

Forward pass.

Parameters:

x (Tensor)

Return type:

Tensor