Ejemplo n.º 1
0
 def __init__(self):
     super(RMSE_DICE_LOSS, self).__init__()
     self.rmse_loss = rmse_loss()
     self.dice_loss = DiceLoss()
Ejemplo n.º 2
0
 def __init__(self):
     super(RmseBceDiceLoss, self).__init__()
     self.rmse = rmse_loss()
     self.bce_dice = BCEDiceLoss()
Ejemplo n.º 3
0
 def __init__(self):
     super(RMSE_SSIM_LOSS, self).__init__()
     self.bce_loss = rmse_loss()
     self.ssim_loss = SSIM_Loss()
Ejemplo n.º 4
0
 def __init__(self):
     super(BCE_RMSE_LOSS, self).__init__()
     self.bce_loss = bcewithlogits_loss()
     self.rmse_loss = rmse_loss()