Skip to content

Commit db2f65b

Browse files
Add logging for case processing in concatenate_modalities function
1 parent 188fa53 commit db2f65b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

monai/nvflare/utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,7 @@ def concatenate_modalities(dataset_format,data_dir, modality_dict, output_data_d
8585
for modality_id in modality_dict:
8686
if modality_id != "label":
8787
data[modality_id] = str(Path(data_dir).joinpath("imagesTr").joinpath(case_id + modality_dict[modality_id]))
88+
print(f"Processing case: {case_id}")
8889
transform(data)
8990
shutil.copy(Path(data_dir).joinpath(f"labelsTr/{case_id}"+ modality_dict[modality_id]), Path(labels_output_dir).joinpath(f"{case_id}"+ modality_dict[modality_id]))
9091
elif dataset_format == "subfolders":
@@ -107,6 +108,7 @@ def concatenate_modalities(dataset_format,data_dir, modality_dict, output_data_d
107108
data[modality_id] = str(Path(data_dir).joinpath(patient_id, patient_id + modality_dict[modality_id]))
108109
else:
109110
data[modality_id] = str(Path(data_dir).joinpath(patient_id, modality_dict[modality_id]))
111+
print(f"Processing case: {case_id}")
110112
transform(data)
111113

112114
def prepare_data_folder_api(data_dir,

0 commit comments

Comments
 (0)