diff --git a/src/main/java/org/bioimageanalysis/icy/deeplearning/tensorflow/v1/Tensorflow1Interface.java b/src/main/java/org/bioimageanalysis/icy/deeplearning/tensorflow/v1/Tensorflow1Interface.java index 6ed3e29ab43039ae17baf232bc0bf9155bf62bf4..1e2506c8c23336a12f82780fd87b3590c8b92547 100644 --- a/src/main/java/org/bioimageanalysis/icy/deeplearning/tensorflow/v1/Tensorflow1Interface.java +++ b/src/main/java/org/bioimageanalysis/icy/deeplearning/tensorflow/v1/Tensorflow1Interface.java @@ -240,6 +240,7 @@ public class Tensorflow1Interface implements DeepLearningEngineInterface public void runInterprocessing(List<Tensor<?>> inputTensors, List<Tensor<?>> outputTensors) throws RunModelException { createTensorsForInterprocessing(inputTensors); + createTensorsForInterprocessing(outputTensors); List<String> args = getProcessCommandsWithoutArgs(); args.add(modelFolder); args.add(this.tmpDir); diff --git a/src/main/java/org/bioimageanalysis/icy/deeplearning/tensorflow/v1/tensor/MappedBufferToImgLib2.java b/src/main/java/org/bioimageanalysis/icy/deeplearning/tensorflow/v1/tensor/MappedBufferToImgLib2.java index f6c84c8163e484c443f404364eb94e22dec3746d..2609986f5f0d7cb3106284ea7ea51c1078b3f07d 100644 --- a/src/main/java/org/bioimageanalysis/icy/deeplearning/tensorflow/v1/tensor/MappedBufferToImgLib2.java +++ b/src/main/java/org/bioimageanalysis/icy/deeplearning/tensorflow/v1/tensor/MappedBufferToImgLib2.java @@ -261,7 +261,7 @@ public final class MappedBufferToImgLib2 long[] shape = new long[0]; if (!shapeStr.isEmpty() && !shapeStr.equals("[]")) { - shapeStr = shapeStr.substring(0, shapeStr.length() - 1); + shapeStr = shapeStr.substring(1, shapeStr.length() - 1); String[] tokens = shapeStr.split(", ?"); shape = Arrays.stream(tokens).mapToLong(Long::parseLong).toArray(); }