Add cores param

CBICA · Oct 9, 2024 · 00a2610 · 00a2610
1 parent 0820179
commit 00a2610
Show file tree

Hide file tree

Showing 5 changed files with 476 additions and 1 deletion.
diff --git a/NiChart_DLMUSE/__main__.py b/NiChart_DLMUSE/__main__.py
@@ -75,6 +75,15 @@ def main() -> None:
         required=True,
     )
 
+    parser.add_argument(
+        "-c",
+        "--cores",
+        type=str,
+        help="Number of cores",
+        default=4,
+        required=False,
+    )
+
     # VERSION argument
     help = "Show the version and exit"
     parser.add_argument(
@@ -112,7 +121,7 @@ def main() -> None:
         os.system("DLMUSE --clear_cache")
 
     # Run pipeline
-    no_threads = 4  # for now
+    no_threads = args.cores # for now
     subfolders = split_data(in_data, no_threads)
 
     threads = []

diff --git a/output/split_2/temp_working_dir/s2_dlicv/dataset.json b/output/split_2/temp_working_dir/s2_dlicv/dataset.json
@@ -0,0 +1,11 @@
+{
+    "channel_names": {
+        "0": "MRI"
+    },
+    "file_ending": ".nii.gz",
+    "labels": {
+        "background": 0,
+        "class1": 1
+    },
+    "numTraining": 1806
+}