From a38483c8478a750e8a3f5567e57d945379e077ab Mon Sep 17 00:00:00 2001 From: Jun <1259389904@qq.com> Date: Wed, 31 Mar 2021 08:50:16 +0800 Subject: [PATCH] Delete nnUNetTrainer_DiceHDBinary.py --- test/network_training/nnUNetTrainer_DiceHDBinary.py | 10 ---------- 1 file changed, 10 deletions(-) delete mode 100644 test/network_training/nnUNetTrainer_DiceHDBinary.py diff --git a/test/network_training/nnUNetTrainer_DiceHDBinary.py b/test/network_training/nnUNetTrainer_DiceHDBinary.py deleted file mode 100644 index 61ad4d5..0000000 --- a/test/network_training/nnUNetTrainer_DiceHDBinary.py +++ /dev/null @@ -1,10 +0,0 @@ -from nnunet.training.loss_functions.boundary_loss import DC_and_HDBinary_loss -from nnunet.training.network_training.nnUNetTrainer import nnUNetTrainer - - -class nnUNetTrainer_DiceHDBinary(nnUNetTrainer): - def __init__(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None, - unpack_data=True, deterministic=True, fp16=False): - super().__init__(plans_file, fold, output_folder, dataset_directory, batch_dice, stage, - unpack_data, deterministic, fp16) - self.loss = DC_and_HDBinary_loss({'batch_dice': self.batch_dice, 'smooth': 1e-5, 'do_bg': False, 'square': False},{})