From 753691bcea1b6745d8ac74e5469287daec26b219 Mon Sep 17 00:00:00 2001 From: Xiao Furen Date: Mon, 3 Feb 2025 07:56:27 +0800 Subject: [PATCH] Need NFS to lock --- src/g4synthmorph.py | 5 +++-- src/m6synthmorph.py | 14 +++++++------- 2 files changed, 10 insertions(+), 9 deletions(-) diff --git a/src/g4synthmorph.py b/src/g4synthmorph.py index 37ada25..92c9e60 100644 --- a/src/g4synthmorph.py +++ b/src/g4synthmorph.py @@ -47,8 +47,9 @@ from mri_synthmorph.synthmorph import registration import surfa as sf -PATIENTS_ROOT = '/mnt/1218/Public/dataset2/G4' -OUT_ROOT = '/mnt/1218/Public/dataset2/G4-synthmorph' +### Need NFS for lock +PATIENTS_ROOT = '/mnt/1220/Public/dataset2/G4' +OUT_ROOT = '/mnt/1220/Public/dataset2/G4-synthmorph' SHELVE = os.path.join(OUT_ROOT, '0shelve') MAX_Y = 256 diff --git a/src/m6synthmorph.py b/src/m6synthmorph.py index 5950097..f723d4a 100644 --- a/src/m6synthmorph.py +++ b/src/m6synthmorph.py @@ -50,8 +50,9 @@ from mri_synthmorph.synthmorph import registration import surfa as sf -PATIENTS_ROOT = '/mnt/1218/Public/dataset2/M6' -OUT_ROOT = '/mnt/1218/Public/dataset2/M6-synthmorph' +### Need NFS for lock +PATIENTS_ROOT = '/mnt/1220/Public/dataset2/M6' +OUT_ROOT = '/mnt/1220/Public/dataset2/M6-synthmorph' SHELVE = os.path.join(OUT_ROOT, '0shelve') MAX_Y = 256 @@ -202,12 +203,10 @@ def register(ct0, ct1, moving, out_root): default['model'] = m default['out_dir'] = out_root/m - # if m in ('affine', 'rigid'): - # default['trans'] = 'trans.lta' - # default['inverse'] = 'inverse.lta' + # if m == 'rigid': + # default['gpu'] = False # else: - # default['trans'] = 'trans.nii.gz' - # default['inverse'] = 'inverse.nii.gz' + # default['gpu'] = True arg=argparse.Namespace(**default) @@ -357,6 +356,7 @@ def main(): EXCLUDE = ( # 'LLUQJUY4', #cervical + '2XYU7UHB', # I tensorflow/core/framework/local_rendezvous.cc:404] Local rendezvous is aborting with status: INVALID_ARGUMENT: Input is not invertible. ) os.makedirs(OUT_ROOT, exist_ok=True)