ENH: FLAIR inpainting.

ANTsX · Feb 12, 2024 · eccd320 · eccd320
1 parent 4379f4a
commit eccd320
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/R/getPretrainedNetwork.R b/R/getPretrainedNetwork.R
@@ -96,6 +96,7 @@ getPretrainedNetwork <- function(
               "inpainting_axial_rmnet_weights",
               "inpainting_axial_rmnet_flair_weights",
               "inpainting_coronal_rmnet_flair_weights",
+              "inpainting_sagittal_rmnet_flair_weights",
               "koniqMBCS",
               "koniqMS",
               "koniqMS2",
@@ -214,6 +215,7 @@ getPretrainedNetwork <- function(
     inpainting_axial_rmnet_weights = "https://figshare.com/ndownloader/files/44244446",
     inpainting_axial_rmnet_flair_weights = "https://figshare.com/ndownloader/files/44406923",
     inpainting_coronal_rmnet_flair_weights = "https://figshare.com/ndownloader/files/44468984",
+    inpainting_sagittal_rmnet_flair_weights = "https://figshare.com/ndownloader/files/44511356",
     koniqMBCS = "https://ndownloader.figshare.com/files/24967376",
     koniqMS = "https://figshare.com/ndownloader/files/35295403",
     koniqMS2 = "https://figshare.com/ndownloader/files/35295397",

diff --git a/R/inpainting.R b/R/inpainting.R
@@ -58,9 +58,9 @@ wholeHeadInpainting <- function( image, roiMask, modality = "t1", mode = "axial"
         { 
         weightsFile <- getPretrainedNetwork( "inpainting_sagittal_rmnet_weights",
                                             antsxnetCacheDirectory = antsxnetCacheDirectory )
-        # } else if( modality == "flair" ) {
-        # weightsFile <- getPretrainedNetwork( "inpainting_sagittal_rmnet_flair_weights",
-        #                                     antsxnetCacheDirectory = antsxnetCacheDirectory )
+        } else if( modality == "flair" ) {
+        weightsFile <- getPretrainedNetwork( "inpainting_sagittal_rmnet_flair_weights",
+                                            antsxnetCacheDirectory = antsxnetCacheDirectory )
         } else {
         stop( "Unrecognized modality." )
         }