Update total_mr/Prediction.yml

total_mr/Prediction.yml

@@ -60,15 +60,14 @@ Predictor:
     subset: None
     filter: None
     dataset_filenames:
-    - ./Dataset/:nii.gz
+    - ./Dataset/:mha
     use_cache: false
     batch_size: 1
   outputs_dataset:
     Head:Softmax:
       OutputDataset:
         name_class: OutSameAsGroupDataset
-        before_reduction_transforms: None
-        after_reduction_transforms: 
+        before_reduction_transforms:
           Softmax:
             dim: 0
           Argmax:
@@ -76,17 +75,19 @@ Predictor:
           TensorCast:
             dtype: uint8
             inverse: true
+        after_reduction_transforms: 
+          Sum:
+            dim: 0
         final_transforms: None
         dataset_filename: Dataset:mha
-        group: Seg
+        group: Output
         same_as_group: Volume:Volume
         patch_combine: Cosinus
-        inverse_transform: true
         reduction: Mean
   train_name: TotalSegmentator
   manual_seed: 32
   gpu_checkpoints: None
   images_log: None
-  combine: Model:Combine
+  combine: Concat
   autocast: false
   data_log: None