from .s2f_trainer import train_s2f, S2FLoss, calculate_soft_dice_loss