Skip to content

Commit

Permalink
swapping the last two modalities to match order Evan used in most recent
Browse files Browse the repository at this point in the history
initial model
  • Loading branch information
brandon-edwards committed Nov 6, 2024
1 parent 4b0d1b0 commit c413bf6
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions examples/fl_post/fl/project/nnunet_data_setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,8 @@

num_to_modality = {'_0000': '_brain_t1n.nii.gz',
'_0001': '_brain_t2w.nii.gz',
'_0002': '_brain_t1c.nii.gz',
'_0003': '_brain_t2f.nii.gz'}
'_0002': '_brain_t2f.nii.gz',
'_0003': '_brain_t1c.nii.gz'}

def get_subdirs(parent_directory):
subjects = os.listdir(parent_directory)
Expand Down Expand Up @@ -418,4 +418,5 @@ def setup_fl_data(postopp_pardir,
print(f"plans_path: {col_paths['plans_path']}")
print(f"\n### ### ### ### ### ### ###\n")

return col_paths
return col_paths

0 comments on commit c413bf6

Please sign in to comment.