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

termiate upgrading tf1 interface to imglib2

parent f1f80574
Branches
No related tags found
No related merge requests found
...@@ -7,15 +7,16 @@ import org.bioimageanalysis.icy.deeplearning.exceptions.LoadModelException; ...@@ -7,15 +7,16 @@ import org.bioimageanalysis.icy.deeplearning.exceptions.LoadModelException;
import org.bioimageanalysis.icy.deeplearning.exceptions.RunModelException; import org.bioimageanalysis.icy.deeplearning.exceptions.RunModelException;
import org.bioimageanalysis.icy.deeplearning.tensor.Tensor; import org.bioimageanalysis.icy.deeplearning.tensor.Tensor;
import org.bioimageanalysis.icy.deeplearning.utils.DeepLearningInterface; import org.bioimageanalysis.icy.deeplearning.utils.DeepLearningInterface;
import org.bioimageanalysis.icy.tensorflow.v1.tensor.Nd4fBuilder; import org.bioimageanalysis.icy.tensorflow.v1.tensor.ImgLib2Builder;
import org.bioimageanalysis.icy.tensorflow.v1.tensor.TensorBuilder; import org.bioimageanalysis.icy.tensorflow.v1.tensor.TensorBuilder;
import org.nd4j.shade.protobuf.InvalidProtocolBufferException;
import org.tensorflow.SavedModelBundle; import org.tensorflow.SavedModelBundle;
import org.tensorflow.Session; import org.tensorflow.Session;
import org.tensorflow.framework.MetaGraphDef; import org.tensorflow.framework.MetaGraphDef;
import org.tensorflow.framework.SignatureDef; import org.tensorflow.framework.SignatureDef;
import org.tensorflow.framework.TensorInfo; import org.tensorflow.framework.TensorInfo;
import com.google.protobuf.InvalidProtocolBufferException;
/** /**
* This plugin includes the libraries to convert back and forth TensorFlow 1 to Sequences and IcyBufferedImages. * This plugin includes the libraries to convert back and forth TensorFlow 1 to Sequences and IcyBufferedImages.
...@@ -120,7 +121,7 @@ public class Tensorflow1Interface implements DeepLearningInterface ...@@ -120,7 +121,7 @@ public class Tensorflow1Interface implements DeepLearningInterface
if (outputNDArrays.size() != outputTensors.size()) if (outputNDArrays.size() != outputTensors.size())
throw new RunModelException(outputNDArrays.size(), outputTensors.size()); throw new RunModelException(outputNDArrays.size(), outputTensors.size());
for (int i = 0; i < outputNDArrays.size(); i ++) { for (int i = 0; i < outputNDArrays.size(); i ++) {
outputTensors.get(i).setData(Nd4fBuilder.build(outputNDArrays.get(i))); outputTensors.get(i).setData(ImgLib2Builder.build(outputNDArrays.get(i)));
} }
return outputTensors; return outputTensors;
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment