From 6af3eeabbc2a26a16cb085810ba505da08e17875 Mon Sep 17 00:00:00 2001 From: Hao-Ting Wang Date: Fri, 31 May 2024 17:23:22 -0400 Subject: [PATCH 1/2] [full_test] fix rerun --- giga_connectome/mask.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/giga_connectome/mask.py b/giga_connectome/mask.py index db0dbbf..b242bac 100644 --- a/giga_connectome/mask.py +++ b/giga_connectome/mask.py @@ -71,6 +71,10 @@ def generate_gm_mask_atlas( nib.save( subject_mask_nii, subject_mask_dir / target_subject_mask_file_name ) + else: + subject_mask_nii = load_img( + subject_mask_dir / target_subject_mask_file_name + ) if not target_subject_seg: subject_seg_niis = resample_atlas_collection( From f413d0307099ad695ad3d3d0c99f9a7d5aced2ce Mon Sep 17 00:00:00 2001 From: Hao-Ting Wang Date: Fri, 31 May 2024 17:35:13 -0400 Subject: [PATCH 2/2] [full_test] fix output of mask/seg checking --- giga_connectome/mask.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/giga_connectome/mask.py b/giga_connectome/mask.py index 4f6ed6d..63fd806 100644 --- a/giga_connectome/mask.py +++ b/giga_connectome/mask.py @@ -87,11 +87,10 @@ def generate_gm_mask_atlas( ) else: subject_seg_niis = [ - load_img(subject_mask_dir / i) - for i in target_subject_seg_file_names + subject_mask_dir / i for i in target_subject_seg_file_names ] - return subject_mask_nii, subject_seg_niis + return subject_mask_dir / target_subject_mask_file_name, subject_seg_niis def generate_subject_gm_mask(