xfr/ck-preprocess
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Need NFS to lock

Browse source
This commit is contained in:
Xiao Furen 2025-02-03 07:56:27 +08:00
parent cdd5b406a6
commit 753691bcea
2 changed files with 10 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/g4synthmorph.py
View file

@ -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

14
src/m6synthmorph.py
View file

@ -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)