Fix locked output segmentation from GeodesicTraining

dboun · May 2, 2019 · 653680f · 653680f
1 parent 49910be
commit 653680f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/core/modules/algorithms/GeodesicTraining/GeodesicTrainingModule.cpp b/core/modules/algorithms/GeodesicTraining/GeodesicTrainingModule.cpp
@@ -255,8 +255,10 @@ void GeodesicTrainingModule::Algorithm()
                 segm = mitk::IOUtil::Load<mitk::LabelSetImage>(tmpPath.toStdString());
                 // EO Fix this hack
 
+                // segm->GetActiveLabelSet()->SetActiveLabel(0);
+                segm->GetActiveLabelSet()->SetAllLabelsLocked(false);
                 segm->GetActiveLabelSet()->SetActiveLabel(0);
-                                
+
                 ds-> AddMitkLabelSetImageToSubject(this->GetUid(), 
                     segm, outputPath + "segmentation.nrrd", "Segmentation", "<Segmentation>   " 
                 );
@@ -368,6 +370,8 @@ void GeodesicTrainingModule::Algorithm()
                 segm = mitk::IOUtil::Load<mitk::LabelSetImage>(tmpPath.toStdString());
                 // EO Fix this hack
 
+                // segm->GetActiveLabelSet()->SetActiveLabel(0);
+                segm->GetActiveLabelSet()->SetAllLabelsLocked(false);
                 segm->GetActiveLabelSet()->SetActiveLabel(0);
 
                 ds-> AddMitkLabelSetImageToSubject(this->GetUid(),