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

adapt to model runner re organization

parent 272b26fd
Branches
No related tags found
No related merge requests found
...@@ -3,12 +3,12 @@ package org.bioimageanalysis.icy.deeplearning.tensorflow.v1; ...@@ -3,12 +3,12 @@ package org.bioimageanalysis.icy.deeplearning.tensorflow.v1;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.List; import java.util.List;
import org.bioimageanalysis.icy.deeplearning.engine.DeepLearningEngineInterface;
import org.bioimageanalysis.icy.deeplearning.exceptions.LoadModelException; 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.tensorflow.v1.tensor.ImgLib2Builder; import org.bioimageanalysis.icy.deeplearning.tensorflow.v1.tensor.ImgLib2Builder;
import org.bioimageanalysis.icy.deeplearning.tensorflow.v1.tensor.TensorBuilder; import org.bioimageanalysis.icy.deeplearning.tensorflow.v1.tensor.TensorBuilder;
import org.bioimageanalysis.icy.deeplearning.utils.DeepLearningInterface;
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;
...@@ -26,7 +26,7 @@ import com.google.protobuf.InvalidProtocolBufferException; ...@@ -26,7 +26,7 @@ import com.google.protobuf.InvalidProtocolBufferException;
* @see TensorBuilder TensorBuilder: Create tensors from images and sequences. * @see TensorBuilder TensorBuilder: Create tensors from images and sequences.
* @author Carlos Garcia Lopez de Haro and Daniel Felipe Gonzalez Obando * @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"}; private static final String[] MODEL_TAGS = {"serve", "inference", "train", "eval", "gpu", "tpu"};
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment