Merge pull request #63 from ilastik/fix-pixel-class-undefined-export-…

…source explicitly setting export_source in any case
ilastik · Jul 28, 2020 · d3e9ec6 · d3e9ec6
2 parents f8b867b + ded017f
commit d3e9ec6
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/main/java/org/ilastik/ilastik4ij/executors/PixelClassification.java b/src/main/java/org/ilastik/ilastik4ij/executors/PixelClassification.java
@@ -30,6 +30,9 @@ protected List<String> buildCommandLine(Map<String, String> tempFiles, PixelPred
         if (pixelPredictionType == PixelPredictionType.Segmentation) {
             commandLine.add("--export_source=Simple Segmentation");
         }
+        else if (pixelPredictionType == PixelPredictionType.Probabilities) {
+            commandLine.add("--export_source=Probabilities");
+        }
         commandLine.add("--raw_data=" + tempFiles.get(rawInputTempFile));
         commandLine.add("--output_filename_format=" + tempFiles.get(outputTempFile));