Skip to content

Commit 6bb20a7

Browse files
committed
dataset processing jupiter
1 parent cf27acf commit 6bb20a7

1 file changed

Lines changed: 15 additions & 7 deletions

File tree

tutorials/tutorial_Dataset_processing.ipynb

Lines changed: 15 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,8 @@
7171
"metadata": {},
7272
"outputs": [],
7373
"source": [
74-
"from TPTBox.core.bids_files import formats,entities_keys\n",
74+
"from TPTBox.core.bids_files import entities_keys, formats\n",
75+
"\n",
7576
"print('Known formats:\\n','\\n'.join(formats))\n",
7677
"print()\n",
7778
"print()\n",
@@ -104,6 +105,7 @@
104105
"outputs": [],
105106
"source": [
106107
"from pathlib import Path\n",
108+
"\n",
107109
"from TPTBox.core.dicom.dicom_extract import extract_dicom_folder\n",
108110
"\n",
109111
"path_to_dicom_dataset = \"TODO\"\n",
@@ -123,6 +125,7 @@
123125
"outputs": [],
124126
"source": [
125127
"from pathlib import Path\n",
128+
"\n",
126129
"path_to_dicom_dataset = \"/media/data/robert/datasets/dicom_example/VR-DICOM/\"\n",
127130
"dataset_name = 'VR-DICOM2'\n",
128131
"target_folder = Path(path_to_dicom_dataset).parent\n",
@@ -142,8 +145,9 @@
142145
"metadata": {},
143146
"outputs": [],
144147
"source": [
145-
"from TPTBox import BIDS_Global_info,BIDS_FILE\n",
148+
"from TPTBox import BIDS_FILE, BIDS_Global_info\n",
146149
"from TPTBox.core.bids_constants import sequence_splitting_keys\n",
150+
"\n",
147151
"print(\"if one of the values of these keys is diffrent, than it is considered a other sequence:\", sequence_splitting_keys)\n",
148152
"print(\"sub will alway split\")\n",
149153
"\n",
@@ -215,10 +219,11 @@
215219
"metadata": {},
216220
"outputs": [],
217221
"source": [
218-
"from TPTBox.stitching import stitching\n",
219-
"from TPTBox import to_nii\n",
220222
"from concurrent.futures import ProcessPoolExecutor\n",
221223
"\n",
224+
"from TPTBox import to_nii\n",
225+
"from TPTBox.stitching import stitching\n",
226+
"\n",
222227
"derivative_folder = \"derivative_stiched\"\n",
223228
"\n",
224229
"def process_files(files):\n",
@@ -269,14 +274,15 @@
269274
"metadata": {},
270275
"outputs": [],
271276
"source": [
272-
"from TPTBox.segmentation import run_totalvibeseg\n",
273277
"from TPTBox import BIDS_FILE\n",
278+
"from TPTBox.segmentation import run_totalvibeseg\n",
279+
"\n",
274280
"# run_totalvibeseg\n",
275281
"# You can alos use a string/Path if you want to set the path yourself.\n",
276282
"dataset = \"/media/data/robert/datasets/dicom_example/dataset-VR-DICOM2/\"\n",
277283
"in_file = BIDS_FILE(f\"{dataset}/derivative_stiched/sub-111168222/T2w/sub-111168222_sequ-301-stiched_acq-ax_part-water_T2w.nii.gz\",dataset)\n",
278284
"out_file = in_file.get_changed_path(\"nii.gz\",\"msk\",parent=\"derivative\",info={\"seg\":\"TotalVibeSegmentator\",\"mod\":in_file.bids_format})\n",
279-
"run_totalvibeseg(in_file,out_file)"
285+
"run_totalvibeseg(in_file,out_file,True)"
280286
]
281287
},
282288
{
@@ -309,7 +315,8 @@
309315
"outputs": [],
310316
"source": [
311317
"# Pick a fitting model:\n",
312-
"from spineps.models import modelid2folder_semantic,modelid2folder_instance\n",
318+
"from spineps.models import modelid2folder_instance, modelid2folder_semantic\n",
319+
"\n",
313320
"print('Available Semantic Models',modelid2folder_semantic())\n",
314321
"print('Available Instance Models',modelid2folder_instance())\n",
315322
"\n",
@@ -331,6 +338,7 @@
331338
"outputs": [],
332339
"source": [
333340
"from TPTBox.segmentation.spineps import run_spineps_single\n",
341+
"\n",
334342
"#With 'ignore_compatibility_issues = True' you can force to rund\n",
335343
"out_paths = run_spineps_single(\n",
336344
" file_path,\n",

0 commit comments

Comments
 (0)