def __init__(self, weight=None, sigmoid_normalization=True):
super().__init__(weight, sigmoid_normalization)
self.smoothing = GaussianSmoothing(1, 7, 2, 3).to('cuda')
def dice(self, input, target, kernel_size, sigma, weight):
self.smoothing = GaussianSmoothing(1, kernel_size, sigma, 3).to('cuda')
g_input = self.smoothing(input, kernel_size)
g_target = self.smoothing(target, kernel_size)
return compute_per_channel_dice(g_input, g_target, weight=self.weight)
def __init__(self, weight=None, sigmoid_normalization=True):
super().__init__(weight, sigmoid_normalization)
# channel, kernel_size, sigma, dim
self.smoothing = GaussianSmoothing(1, 7, 2, 3).to('cuda')