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')