Fix inference YAMLs: pin single R+ACS; document training schedule + how to override

README.md

@@ -54,6 +54,29 @@ Each experiment is a pair (no iteration in the filename):
 | `vsharp_ads_2d_frame` | vSHARP | ADS 2D frame-specific |
 | `medl_ads_2d_frame` | MEDL | ADS 2D frame-specific |
 
+## Training protocol & inference masking
+
+These models were trained on **CMRxRecon** cine with **mixed** discrete
+accelerations (typically `[4.0327, 6, 8.2]`, or init2 variants such as
+`[6, 8.2, 10.25]`) and ACS `center_fractions` of matching length (usually
+`0.04`). Sampling scheme is config-specific (`FastMRIRandom`,
+`FastMRIEquispaced`, or `Gaussian2D`).
+
+Released YAMLs are inference-only and pin **exactly one** `accelerations`
+value and **exactly one** `center_fractions` value (default: validation
+`val-4x`). DIRECT’s `MaskFunc` randomly samples from lists — do **not** put
+multi-R training lists into inference configs.
+
+To evaluate another trained rate, edit `inference.dataset.transforms.masking`
+and keep both lists length 1, e.g.:
+
+```yaml
+masking:
+  name: FastMRIRandom
+  accelerations: [8.2]
+  center_fractions: [0.04]
+```
+
 ## Usage
 
 ```bash

medl_ads_1d.yaml

@@ -60,6 +60,11 @@ inference:
             normalization:
                 scaling_key: acs_kspace
                 scale_percentile: 0.995
-        kspace_key: kspace_full
-        compute_mask: false
+            use_acs_as_mask: true
+            delete_acs_mask: false
+            padding_eps: 0.0
+            augmentation:
+                pad:
+                    - 512
+                    - 246
         kspace_context: time

medl_ads_1d_frame.yaml

@@ -49,7 +49,7 @@ inference:
             masking:
                 name: FastMRIRandom
                 accelerations:
-                    - 8.2
+                    - 4.0327
                 center_fractions:
                     - 0.04
             cropping:
@@ -59,6 +59,11 @@ inference:
             normalization:
                 scaling_key: acs_kspace
                 scale_percentile: 0.995
-        kspace_key: kspace_full
-        compute_mask: false
+            use_acs_as_mask: true
+            delete_acs_mask: false
+            padding_eps: 0.0
+            augmentation:
+                pad:
+                    - 512
+                    - 246
         kspace_context: time

medl_ads_1d_frame_init2.yaml

@@ -60,3 +60,11 @@ inference:
             normalization:
                 scaling_key: acs_kspace
                 scale_percentile: 0.995
+            use_acs_as_mask: false
+            delete_acs_mask: false
+            dynamic_mask: true
+            augmentation:
+                pad:
+                    - 512
+                    - 246
+        kspace_context: time

medl_ads_1d_init2.yaml

@@ -60,5 +60,11 @@ inference:
             normalization:
                 scaling_key: acs_kspace
                 scale_percentile: 0.995
+            use_acs_as_mask: false
+            delete_acs_mask: false
+            padding_eps: 0.0
+            augmentation:
+                pad:
+                    - 512
+                    - 246
         kspace_context: time
-        kspace_key: kspace_full

medl_ads_2d.yaml

@@ -60,3 +60,10 @@ inference:
             normalization:
                 scaling_key: acs_kspace
                 scale_percentile: 0.995
+            use_acs_as_mask: true
+            delete_acs_mask: false
+            augmentation:
+                pad:
+                    - 512
+                    - 246
+        kspace_context: time

medl_ads_2d_frame.yaml

@@ -41,6 +41,7 @@ physics:
     backward_operator: ifft2
 inference:
     batch_size: 1
+    crop: header
     dataset:
         name: CMRxRecon
         transforms:
@@ -59,3 +60,10 @@ inference:
             normalization:
                 scaling_key: acs_kspace
                 scale_percentile: 0.995
+            use_acs_as_mask: true
+            delete_acs_mask: false
+            augmentation:
+                pad:
+                    - 512
+                    - 246
+        kspace_context: time

vsharp_ads_1d.yaml

@@ -75,6 +75,11 @@ inference:
             normalization:
                 scaling_key: acs_kspace
                 scale_percentile: 0.995
-        kspace_key: kspace_full
-        compute_mask: false
+            use_acs_as_mask: true
+            delete_acs_mask: false
+            padding_eps: 0.0
+            augmentation:
+                pad:
+                    - 512
+                    - 246
         kspace_context: time

vsharp_ads_1d_frame.yaml

@@ -64,7 +64,7 @@ inference:
             masking:
                 name: FastMRIRandom
                 accelerations:
-                    - 8.2
+                    - 4.0327
                 center_fractions:
                     - 0.04
             cropping:
@@ -74,6 +74,11 @@ inference:
             normalization:
                 scaling_key: acs_kspace
                 scale_percentile: 0.995
-        kspace_key: kspace_full
-        compute_mask: false
+            use_acs_as_mask: true
+            delete_acs_mask: false
+            padding_eps: 0.0
+            augmentation:
+                pad:
+                    - 512
+                    - 246
         kspace_context: time

vsharp_ads_1d_frame_init2.yaml

@@ -75,3 +75,11 @@ inference:
             normalization:
                 scaling_key: acs_kspace
                 scale_percentile: 0.995
+            use_acs_as_mask: false
+            delete_acs_mask: false
+            dynamic_mask: true
+            augmentation:
+                pad:
+                    - 512
+                    - 246
+        kspace_context: time

vsharp_ads_1d_init2.yaml

@@ -75,5 +75,11 @@ inference:
             normalization:
                 scaling_key: acs_kspace
                 scale_percentile: 0.995
+            use_acs_as_mask: false
+            delete_acs_mask: false
+            padding_eps: 0.0
+            augmentation:
+                pad:
+                    - 512
+                    - 246
         kspace_context: time
-        kspace_key: kspace_full

vsharp_ads_2d.yaml

@@ -75,3 +75,10 @@ inference:
             normalization:
                 scaling_key: acs_kspace
                 scale_percentile: 0.995
+            use_acs_as_mask: true
+            delete_acs_mask: false
+            augmentation:
+                pad:
+                    - 512
+                    - 246
+        kspace_context: time

vsharp_ads_2d_frame.yaml

@@ -56,6 +56,7 @@ physics:
     backward_operator: ifft2
 inference:
     batch_size: 1
+    crop: header
     dataset:
         name: CMRxRecon
         transforms:
@@ -74,3 +75,10 @@ inference:
             normalization:
                 scaling_key: acs_kspace
                 scale_percentile: 0.995
+            use_acs_as_mask: true
+            delete_acs_mask: false
+            augmentation:
+                pad:
+                    - 512
+                    - 246
+        kspace_context: time