From d02d62cbbd44a689dd94a343cbe922cf9d8d1be3 Mon Sep 17 00:00:00 2001 From: carlosuc3m <100329787@alumnos.uc3m.es> Date: Thu, 2 Feb 2023 13:22:56 +0100 Subject: [PATCH] adapt to model runner re organization --- .../icy/deeplearning/tensorflow/v1/Tensorflow1Interface.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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 1e40203..c50a3fc 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 @@ -3,12 +3,12 @@ package org.bioimageanalysis.icy.deeplearning.tensorflow.v1; import java.util.ArrayList; import java.util.List; +import org.bioimageanalysis.icy.deeplearning.engine.DeepLearningEngineInterface; 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.tensorflow.SavedModelBundle; import org.tensorflow.Session; import org.tensorflow.framework.MetaGraphDef; @@ -26,7 +26,7 @@ import com.google.protobuf.InvalidProtocolBufferException; * @see TensorBuilder TensorBuilder: Create tensors from images and sequences. * @author Carlos Garcia Lopez de Haro and Daniel Felipe Gonzalez Obando */ -public class Tensorflow1Interface implements DeepLearningInterface +public class Tensorflow1Interface implements DeepLearningEngineInterface { private static final String[] MODEL_TAGS = {"serve", "inference", "train", "eval", "gpu", "tpu"}; -- GitLab