voxelmorph.nn.losses

Loss functions for image registration.

Dice

N-D dice for segmentation

Grad

Grad(penalty='l1', loss_mult=None)

N-D gradient loss.

Source code in voxelmorph/nn/losses.py

def __init__(self, penalty='l1', loss_mult=None):
    self.penalty = penalty
    self.loss_mult = loss_mult

MSE

Mean squared error loss.

NCC

NCC(win=None)

Local (over window) normalized cross correlation loss.

Source code in voxelmorph/nn/losses.py

def __init__(self, win=None):
    self.win = win