Update MR_CBCT/Prediction.yml

MR_CBCT/Prediction.yml

@@ -8,6 +8,26 @@ Predictor:
     groups_src:
       Volume_0:
         groups_dest:
+          MASK:
+            transforms:
+              KonfAIInference:
+                repo_id: VBoussot/ImpactSeg
+                model_name: body
+                checkpoints_name:
+                - Body.pt
+                number_of_tta: 0
+                number_of_mc_dropout: 0
+                per_channel: false
+              ResampleToResolution:
+                spacing:
+                - 1
+                - 1
+                - 3
+                inverse: false
+              Save:
+                dataset: ./Dataset:mha
+            patch_transforms: None
+            is_input: false
           Volume:
             transforms:
               TensorCast/0:
@@ -78,6 +98,9 @@ Predictor:
           TensorCast:
             dtype: int16
             inverse: false
+          Mask:
+            path: MASK
+            value_outside: -1024
         after_reduction_transforms: 
           InferenceStack:
             dataset: Predictions/ImpactSynth/Output:mha