diff --git a/monai/transforms/io/array.py b/monai/transforms/io/array.py index 5f236452ce..eb8aff0656 100644 --- a/monai/transforms/io/array.py +++ b/monai/transforms/io/array.py @@ -223,6 +223,7 @@ class SaveImage(Transform): output_dir: /output, data_root_dir: /foo/bar, output will be: /output/test1/image/image_seg.nii.gz + print_log: whether to print log about the saved file path, etc. default to `True`. """ @@ -240,6 +241,7 @@ def __init__( save_batch: bool = False, squeeze_end_dims: bool = True, data_root_dir: str = "", + print_log: bool = True, ) -> None: self.saver: Union[NiftiSaver, PNGSaver] if output_ext in (".nii.gz", ".nii"): @@ -254,6 +256,7 @@ def __init__( output_dtype=output_dtype, squeeze_end_dims=squeeze_end_dims, data_root_dir=data_root_dir, + print_log=print_log, ) elif output_ext == ".png": self.saver = PNGSaver( @@ -264,6 +267,7 @@ def __init__( mode=InterpolateMode(mode), scale=scale, data_root_dir=data_root_dir, + print_log=print_log, ) else: raise ValueError(f"unsupported output extension: {output_ext}.") diff --git a/monai/transforms/io/dictionary.py b/monai/transforms/io/dictionary.py index 82ee26d4f1..39649e3858 100644 --- a/monai/transforms/io/dictionary.py +++ b/monai/transforms/io/dictionary.py @@ -199,6 +199,7 @@ class SaveImaged(MapTransform): output_dir: /output, data_root_dir: /foo/bar, output will be: /output/test1/image/image_seg.nii.gz + print_log: whether to print log about the saved file path, etc. default to `True`. """ @@ -220,6 +221,7 @@ def __init__( allow_missing_keys: bool = False, squeeze_end_dims: bool = True, data_root_dir: str = "", + print_log: bool = True, ) -> None: super().__init__(keys, allow_missing_keys) self.meta_keys = ensure_tuple_rep(meta_keys, len(self.keys)) @@ -237,6 +239,7 @@ def __init__( save_batch=save_batch, squeeze_end_dims=squeeze_end_dims, data_root_dir=data_root_dir, + print_log=print_log, ) def __call__(self, data):