diff --git a/nnunetv2/training/data_augmentation/custom_transforms/manipulating_data_dict.py b/nnunetv2/training/data_augmentation/custom_transforms/manipulating_data_dict.py deleted file mode 100644 index 587acd721..000000000 --- a/nnunetv2/training/data_augmentation/custom_transforms/manipulating_data_dict.py +++ /dev/null @@ -1,10 +0,0 @@ -from batchgenerators.transforms.abstract_transforms import AbstractTransform - - -class RemoveKeyTransform(AbstractTransform): - def __init__(self, key_to_remove: str): - self.key_to_remove = key_to_remove - - def __call__(self, **data_dict): - _ = data_dict.pop(self.key_to_remove, None) - return data_dict diff --git a/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py b/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py index c5b144a23..c709b6de6 100644 --- a/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py +++ b/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py @@ -50,12 +50,6 @@ from nnunetv2.inference.sliding_window_prediction import compute_gaussian from nnunetv2.paths import nnUNet_preprocessed, nnUNet_results from nnunetv2.training.data_augmentation.compute_initial_patch_size import get_patch_size -from nnunetv2.training.data_augmentation.custom_transforms.cascade_transforms import ApplyRandomBinaryOperatorTransform, \ - RemoveRandomConnectedComponentFromOneHotEncodingTransform -from nnunetv2.training.data_augmentation.custom_transforms.region_based_training import \ - ConvertSegmentationToRegionsTransform -from nnunetv2.training.data_augmentation.custom_transforms.transforms_for_dummy_2d import Convert2DTo3DTransform, \ - Convert3DTo2DTransform from nnunetv2.training.dataloading.data_loader_2d import nnUNetDataLoader2D from nnunetv2.training.dataloading.data_loader_3d import nnUNetDataLoader3D from nnunetv2.training.dataloading.nnunet_dataset import nnUNetDataset