regulariser

Demons Regulariser

class airlab.regulariser.demons.GaussianRegulariser(pixel_spacing, sigma, dtype=torch.float32, device='cpu')

regularise(data)

Graph Diffusion Regulariser

class airlab.regulariser.demons.GraphDiffusionRegulariser(image_size, pixel_spacing, edge_updater, phi=1, dtype=torch.float32, device='cpu')

get_edge_image()

regularise(data)

set_krylov_dim(krylov_dim)

class airlab.regulariser.demons.EdgeUpdaterDisplacementIntensities(pixel_spacing, image, edge_window=0.9, edge_mean=False)

update(data)

class airlab.regulariser.demons.EdgeUpdaterIntensities(pixel_spacing, image, scale=1, edge_window=0.9, edge_mean=False)

set_scale(sale)

update(data)

Displacement Regulariser

class airlab.regulariser.displacement.DiffusionRegulariser(pixel_spacing, size_average=True, reduce=True)

forward(displacement)

class airlab.regulariser.displacement.IsotropicTVRegulariser(pixel_spacing, size_average=True, reduce=True)

forward(displacement)

class airlab.regulariser.displacement.SparsityRegulariser(size_average=True, reduce=True)

forward(displacement)

class airlab.regulariser.displacement.TVRegulariser(pixel_spacing, size_average=True, reduce=True)

forward(displacement)

Parameter Regulariser

class airlab.regulariser.parameter.DiffusionRegulariser(pixel_spacing, size_average=True, reduce=True)

forward(displacement)

class airlab.regulariser.parameter.IsotropicTVRegulariser(parameter_name, scaling=[1], size_average=True, reduce=True)

forward(parameters)

class airlab.regulariser.parameter.SparsityRegulariser(parameter_name, size_average=True, reduce=True)

forward(parameters)

class airlab.regulariser.parameter.TVRegulariser(parameter_name, scaling=[1], size_average=True, reduce=True)

forward(parameters)