From d23b61f5c87a694e70a2b2ef059e3cc93a8f749f Mon Sep 17 00:00:00 2001
From: carlosuc3m <100329787@alumnos.uc3m.es>
Date: Tue, 12 Jul 2022 19:16:10 +0200
Subject: [PATCH] termiate upgrading tf1 interface to imglib2
---
src/main/java/tensorflow/v1/Tensorflow1Interface.java | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/src/main/java/tensorflow/v1/Tensorflow1Interface.java b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
index a0af331..bb68bb7 100644
--- a/src/main/java/tensorflow/v1/Tensorflow1Interface.java
+++ b/src/main/java/tensorflow/v1/Tensorflow1Interface.java
@@ -7,15 +7,16 @@ 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.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.nd4j.shade.protobuf.InvalidProtocolBufferException;
import org.tensorflow.SavedModelBundle;
import org.tensorflow.Session;
import org.tensorflow.framework.MetaGraphDef;
import org.tensorflow.framework.SignatureDef;
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.
@@ -120,7 +121,7 @@ public class Tensorflow1Interface implements DeepLearningInterface
if (outputNDArrays.size() != outputTensors.size())
throw new RunModelException(outputNDArrays.size(), outputTensors.size());
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;
}
--
GitLab