RF: Ensure clipping of anatomical template, simplify workflow

Author: mgxd

nibabies/workflows/anatomical/fit.py

@@ -567,19 +567,11 @@ def init_infant_anat_fit_wf(
 
         if not t1w_preproc:
             LOGGER.info('ANAT Skipping skull-strip, INU-correction only')
-            t1w_n4_only_wf = init_n4_only_wf(
-                omp_nthreads=omp_nthreads,
-                atropos_use_random_seed=not skull_strip_fixed_seed,
-                bids_suffix='T1w',
-                name='t1w_n4_only_wf',
-            )
+            t1w_n4_wf = init_anat_preproc_wf(name='t1w_n4_wf')
             workflow.connect([
-                (t1w_validate, t1w_n4_only_wf, [('out_file', 'inputnode.in_files')]),
-                (t1w_n4_only_wf, t1w_buffer, [
-                    (('outputnode.bias_corrected', pop_file), 't1w_preproc'),
-                ]),
-                (t1w_n4_only_wf, apply_t1w_mask, [
-                    (('outputnode.out_file', pop_file), 'in_file')]),
+                (t1w_validate, t1w_n4_wf, [('out_file', 'inputnode.in_anat')]),
+                (t1w_n4_wf, t1w_buffer, [('outputnode.anat_preproc', 't1w_preproc')]),
+                (t1w_n4_wf, apply_t1w_mask, [('outputnode.anat_preproc', 'in_file')]),
             ])  # fmt:skip
         else:
             LOGGER.info('ANAT Skipping T1w masking')
@@ -719,19 +711,11 @@ def init_infant_anat_fit_wf(
 
         if not t2w_preproc:
             LOGGER.info('ANAT Skipping skull-strip, INU-correction only')
-            t2w_n4_only_wf = init_n4_only_wf(
-                omp_nthreads=omp_nthreads,
-                atropos_use_random_seed=not skull_strip_fixed_seed,
-                bids_suffix='T2w',
-                name='t2w_n4_only_wf',
-            )
+            t2w_n4_wf = init_anat_preproc_wf(name='t2w_n4_wf')
             workflow.connect([
-                (t2w_validate, t2w_n4_only_wf, [('out_file', 'inputnode.in_files')]),
-                (t2w_n4_only_wf, t2w_buffer, [
-                    (('outputnode.bias_corrected', pop_file), 't2w_preproc'),
-                ]),
-                (t2w_n4_only_wf, apply_t2w_mask, [
-                    (('outputnode.out_file', pop_file), 'in_file')]),
+                (t2w_validate, t2w_n4_wf, [('out_file', 'inputnode.in_anat')]),
+                (t2w_n4_wf, t2w_buffer, [('outputnode.anat_preproc', 't2w_preproc')]),
+                (t2w_n4_wf, apply_t2w_mask, [('outputnode.anat_preproc', 'in_file')]),
             ])  # fmt:skip
         else:
             LOGGER.info('ANAT Skipping T2w masking')
@@ -1647,18 +1631,11 @@ def init_infant_single_anat_fit_wf(
 
         if not anat_preproc:
             LOGGER.info('ANAT Skipping skull-strip, INU-correction only')
-            n4_only_wf = init_n4_only_wf(
-                omp_nthreads=omp_nthreads,
-                atropos_use_random_seed=not skull_strip_fixed_seed,
-                bids_suffix=reference_anat,
-                name='n4_only_wf',
-            )
+            anat_n4_wf = init_anat_preproc_wf(name='anat_n4_wf')
             workflow.connect([
-                (anat_validate, n4_only_wf, [('out_file', 'inputnode.in_files')]),
-                (n4_only_wf, anat_buffer, [
-                    (('outputnode.bias_corrected', pop_file), 'anat_preproc'),
-                ]),
-                (n4_only_wf, apply_mask, [(('outputnode.out_file', pop_file), 'in_file')]),
+                (anat_validate, anat_n4_wf, [('out_file', 'inputnode.in_anat')]),
+                (anat_n4_wf, anat_buffer, [('outputnode.anat_preproc', 'anat_preproc')]),
+                (anat_n4_wf, apply_mask, [('outputnode.anat_preproc', 'in_file')]),
             ])  # fmt:skip
         else:
             LOGGER.info(f'ANAT Skipping {reference_anat} masking')

nibabies/workflows/anatomical/preproc.py

@@ -12,11 +12,9 @@ def init_anat_preproc_wf(
 
     This workflow accepts T1w/T2w images as inputs (either raw or a merged template) and performs:
     - Intensity clipping
-    - Denoising
     - N4 Bias Field Correction
 
-    The outputs of this workflow will be used to either create the brainmask,
-    or reconstruct the cortical surfaces.
+    The outputs of this workflow will be a structural reference used for subsequent processing.
 
     Inputs
     ------
@@ -45,7 +43,6 @@ def init_anat_preproc_wf(
     # validate image
     validate = pe.Node(ValidateImage(), name='anat_validate', run_without_submitting=True)
     clip = pe.Node(IntensityClip(p_min=10.0, p_max=99.5), name='clip')
-    # denoise = pe.Node(DenoiseImage(dimension=3, noise_model="Rician"), name="denoise")
     n4_correct = pe.Node(
         N4BiasFieldCorrection(
             dimension=3,
@@ -61,14 +58,11 @@ def init_anat_preproc_wf(
     )
     final_clip = pe.Node(IntensityClip(p_min=5.0, p_max=99.5), name='final_clip')
 
-    # fmt:off
     wf.connect([
         (inputnode, validate, [('in_anat', 'in_file')]),
         (validate, clip, [('out_file', 'in_file')]),
         (clip, n4_correct, [('out_file', 'input_image')]),
-        # (denoise, n4_correct, [("output_image", "input_image")]),
         (n4_correct, final_clip, [('output_image', 'in_file')]),
         (final_clip, outputnode, [('out_file', 'anat_preproc')]),
-    ])
-    # fmt:on
+    ])  # fmt:skip
     return wf