tmp

qupath · Jan 8, 2025 · 6c07c62 · 6c07c62
1 parent 4691b23
commit 6c07c62
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/main/java/qupath/ext/instanseg/core/InstanSegOutputToObjectConverter.java b/src/main/java/qupath/ext/instanseg/core/InstanSegOutputToObjectConverter.java
@@ -158,7 +158,11 @@ public List<PathObject> convertToObjects(Parameters<Mat, Mat> params, Mat[] outp
     }
 
     private static void handleAuxOutput(PathObject pathObject, double[] values, BioimageIoSpec.OutputTensor outputTensor) {
-        List<String> outputClasses = List.of(); // todo
+        List<String> outputClasses = new ArrayList<>(); // todo: get from RDF
+        int nClasses = outputTensor.getShape().getShape()[2];
+        for (int i = 0; i < nClasses; i++) {
+            outputClasses.add("Class" + i);
+        }
         if (values == null)
             return;
         var outputType = InstanSegModel.OutputType.valueOf(outputTensor.getName().toUpperCase());

diff --git a/src/main/java/qupath/ext/instanseg/ui/InstanSegTask.java b/src/main/java/qupath/ext/instanseg/ui/InstanSegTask.java
@@ -62,6 +62,7 @@ protected Void call() {
             return null;
         }
         // TODO: HANDLE OUTPUT CHANNELS!
+        // todo: Unclear what this means
         int nOutputs = model.getOutputChannels().orElse(1);
         int[] outputChannels = new int[0];
         if (nOutputs <= 0) {