cellmap_data.transforms.augment.random_gamma

cellmap_data.transforms.augment.random_gamma#

Classes

RandomGamma([gamma_range])

Initialize the random gamma augmentation.

class cellmap_data.transforms.augment.random_gamma.RandomGamma(gamma_range: Sequence[float] = (0.5, 1.5))[source]#

Initialize the random gamma augmentation.

Parameters:

gamma_range (tuple, optional) – Gamma range. Defaults to (0.5, 1.5).

forward(x: Tensor) Tensor[source]#

Forward pass.

Parameters:

x (Tensor)

Return type:

Tensor