diff --git a/monai/data/nifti_saver.py b/monai/data/nifti_saver.py index cacbcc06cd..8c207b0a2c 100644 --- a/monai/data/nifti_saver.py +++ b/monai/data/nifti_saver.py @@ -44,6 +44,7 @@ def __init__( output_dtype: DtypeLike = np.float32, squeeze_end_dims: bool = True, data_root_dir: str = "", + print_log: bool = True, ) -> None: """ Args: @@ -78,6 +79,7 @@ def __init__( 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 NIfTI file path, etc. default to `True`. """ self.output_dir = output_dir @@ -92,6 +94,7 @@ def __init__( self._data_index = 0 self.squeeze_end_dims = squeeze_end_dims self.data_root_dir = data_root_dir + self.print_log = print_log def save(self, data: Union[torch.Tensor, np.ndarray], meta_data: Optional[Dict] = None) -> None: """ @@ -153,7 +156,9 @@ def save(self, data: Union[torch.Tensor, np.ndarray], meta_data: Optional[Dict] dtype=self.dtype, output_dtype=self.output_dtype, ) - print(f"file written: {path}.") + + if self.print_log: + print(f"file written: {path}.") def save_batch(self, batch_data: Union[torch.Tensor, np.ndarray], meta_data: Optional[Dict] = None) -> None: """ diff --git a/monai/data/png_saver.py b/monai/data/png_saver.py index 880f6b204f..90c1acfc1f 100644 --- a/monai/data/png_saver.py +++ b/monai/data/png_saver.py @@ -37,6 +37,7 @@ def __init__( mode: Union[InterpolateMode, str] = InterpolateMode.NEAREST, scale: Optional[int] = None, data_root_dir: str = "", + print_log: bool = True, ) -> None: """ Args: @@ -59,6 +60,7 @@ def __init__( output_dir: /output, data_root_dir: /foo/bar, output will be: /output/test1/image/image_seg.png + print_log: whether to print log about the saved PNG file path, etc. default to `True`. """ self.output_dir = output_dir @@ -68,6 +70,7 @@ def __init__( self.mode: InterpolateMode = InterpolateMode(mode) self.scale = scale self.data_root_dir = data_root_dir + self.print_log = print_log self._data_index = 0 @@ -122,6 +125,9 @@ def save(self, data: Union[torch.Tensor, np.ndarray], meta_data: Optional[Dict] scale=self.scale, ) + if self.print_log: + print(f"file written: {path}.") + def save_batch(self, batch_data: Union[torch.Tensor, np.ndarray], meta_data: Optional[Dict] = None) -> None: """Save a batch of data into png format files.