Skip to content
Snippets Groups Projects
Commit 86bf8086 authored by carlosuc3m's avatar carlosuc3m
Browse files

rearange packages to keep an order within the dl-model-runner projects

parent e1f03523
No related branches found
No related tags found
No related merge requests found
......@@ -108,7 +108,7 @@
<properties>
<package-name>org.bioimageanalysis.icy.deeplearning.tensorflow.v1</package-name>
<main-class>org.bioimageanalysis.icy.deeplearning.tensorflow.v1.TensorFlowInterface</main-class>
<main-class>org.bioimageanalysis.icy.deeplearning.tensorflow.v1.TensorFlow1Interface</main-class>
<license.licenseName>BSD-3-Clause</license.licenseName>
<license.copyrightOwners>Institut Pasteur</license.copyrightOwners>
</properties>
......
package tensorflow.v1;
package org.bioimageanalysis.icy.deeplearning.tensorflow.v1;
import java.util.ArrayList;
import java.util.List;
......@@ -6,9 +6,9 @@ import java.util.List;
import org.bioimageanalysis.icy.deeplearning.exceptions.LoadModelException;
import org.bioimageanalysis.icy.deeplearning.exceptions.RunModelException;
import org.bioimageanalysis.icy.deeplearning.tensor.Tensor;
import org.bioimageanalysis.icy.deeplearning.tensorflow.v1.tensor.ImgLib2Builder;
import org.bioimageanalysis.icy.deeplearning.tensorflow.v1.tensor.TensorBuilder;
import org.bioimageanalysis.icy.deeplearning.utils.DeepLearningInterface;
import org.bioimageanalysis.icy.tensorflow.v1.tensor.ImgLib2Builder;
import org.bioimageanalysis.icy.tensorflow.v1.tensor.TensorBuilder;
import org.tensorflow.SavedModelBundle;
import org.tensorflow.Session;
import org.tensorflow.framework.MetaGraphDef;
......@@ -78,7 +78,7 @@ public class Tensorflow1Interface implements DeepLearningInterface
}
@Override
public List<Tensor<?>> run(List<Tensor<?>> inputTensors, List<Tensor<?>> outputTensors) throws RunModelException {
public void run(List<Tensor<?>> inputTensors, List<Tensor<?>> outputTensors) throws RunModelException {
Session session = model.session();
Session.Runner runner = session.runner();
List<String> inputListNames = new ArrayList<String>();
......@@ -103,7 +103,6 @@ public class Tensorflow1Interface implements DeepLearningInterface
for (org.tensorflow.Tensor<?> tt : resultPatchTensors) {
tt.close();
}
return outputTensors;
}
/**
......
package org.bioimageanalysis.icy.tensorflow.v1.tensor;
package org.bioimageanalysis.icy.deeplearning.tensorflow.v1.tensor;
import java.nio.ByteBuffer;
import java.nio.DoubleBuffer;
......
package org.bioimageanalysis.icy.tensorflow.v1.tensor;
package org.bioimageanalysis.icy.deeplearning.tensorflow.v1.tensor;
import java.nio.ByteBuffer;
import java.nio.DoubleBuffer;
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment