better checking for 2d images.

the assumption is that 3D means X,Y,Z!
knime-ip · Aug 29, 2016 · 223f262 · 223f262
1 parent 6d386b5
commit 223f262
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...nime.knip.ilastik/src/org/knime/knip/ilastik/nodes/headless/IlastikHeadlessNodeModel.java b/...nime.knip.ilastik/src/org/knime/knip/ilastik/nodes/headless/IlastikHeadlessNodeModel.java
@@ -743,8 +743,9 @@ public void setInternalTables(final BufferedDataTable[] tables) {
      */
     protected ImgPlus<T> overrideTimeDimension(final ImgPlus<T> ilastikImg, final ImgPlusValue<?> imgInValue) {
 
-        if (ilastikImg.dimensionIndex(Axes.CHANNEL) != -1 || ilastikImg.numDimensions() == 2) {
-            KNIPGateway.log().error("The ilastik output already has a Channel axis or is 2 dimensional. Skipping override.");
+        if (ilastikImg.dimensionIndex(Axes.CHANNEL) != -1 || imgInValue.getMetadata().dimensionIndex(Axes.Z) == -1) {
+            KNIPGateway.log()
+                    .debug("The ilastik output already has a Channel axis or is 2 dimensional. Skipping override.");
             return ilastikImg;
         }