output filename schema changed

CBICA · Nov 12, 2024 · 2c35c33 · 2c35c33
1 parent 6ff28eb
commit 2c35c33
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 2 deletions.
diff --git a/DLWMLS/__main__.py b/DLWMLS/__main__.py
@@ -120,7 +120,7 @@ def main() -> None:
         "-d",
         type=str,
         required=False,
-        default="002",
+        default="003",
         help="[nnUnet Arg] Dataset with which you would like to predict. You can specify either dataset name or id",
     )
     parser.add_argument(

diff --git a/DLWMLS/__pycache__/__init__.cpython-312.pyc b/DLWMLS/__pycache__/__init__.cpython-312.pyc
diff --git a/DLWMLS/__pycache__/__main__.cpython-312.pyc b/DLWMLS/__pycache__/__main__.cpython-312.pyc
diff --git a/DLWMLS/__pycache__/utils.cpython-312.pyc b/DLWMLS/__pycache__/utils.cpython-312.pyc
diff --git a/DLWMLS/utils.py b/DLWMLS/utils.py
@@ -36,6 +36,7 @@ def rename_and_copy_files(src_folder: str, des_folder: str) -> Tuple[dict, dict]
         new_name = os.path.join(des_folder, rename_file)
         shutil.copy2(old_name, new_name)
         rename_dict[filename] = rename_file
-        rename_back_dict[rename_back] = "DLWMLS_mask_" + filename
+        rename_back_dict[rename_back] = filename.split(".nii")[0] + "_WMLS.nii.gz"
+        print("%s will be renamed to %s" % (old_name, filename.split(".nii")[0] + "_WMLS.nii.gz"))
 
     return rename_dict, rename_back_dict