Update total/Prediction.yml

total/Prediction.yml

@@ -66,15 +66,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:
@@ -82,9 +81,10 @@ Predictor:
           TensorCast:
             dtype: uint8
             inverse: true
+        after_reduction_transforms: None
         final_transforms: None
         dataset_filename: Dataset:mha
-        group: Seg
+        group: Output
         same_as_group: Volume:Volume
         patch_combine: Cosinus
         inverse_transform: true