Commit 0378cc52 authored by Oleg Dzhimiev's avatar Oleg Dzhimiev
Browse files

changed model tag to 'serve'

parent b7a74a73

src/main/java/TensorflowExamplePlugin.java

+26 −4
Original line number Diff line number Diff line
@@ -12,6 +12,7 @@ import org.tensorflow.SavedModelBundle; 
import org.tensorflow.OperationBuilder;

import org.tensorflow.Shape;

import org.tensorflow.Output;

import org.tensorflow.Operation;



import java.util.ArrayList;

import java.util.Collection;
@@ -51,7 +52,9 @@ public class TensorflowExamplePlugin 
{



    public final static String EXPORTDIR = "/home/oleg/GIT/python3-imagej-tiff/data_sets/tf_data_5x5_main_13_heur/exportdir";

    public final static String PB_TAG = "model_pb";

    // tf.saved_model.tag_constants.SERVING = "serve"

    public final static String PB_TAG = "serve";

    



    public static void run()

    {
@@ -155,13 +158,31 @@ public class TensorflowExamplePlugin 
        

        try {



        	//bundle.session().runner().fetch("Disparity_net/stage1done:0").run();

        	

        	

        	// init variable via constant

        	Tensor<Float> t = toTensor2DFloat(rv_stage1_out, tensorsToClose);

        	Output builder_init = bundle.graph().opBuilder("Const", "rv_stage1_out_init").setAttr("dtype", t.dataType()).setAttr("value", t).build().output(0);

        	//Output builder_init = bundle.graph().opBuilder("Const", "rv_stage1_out_init").setAttr("dtype", t.dataType()).setAttr("value", t).build().output(0);

        	

        	// variable

        	OperationBuilder builder2 = bundle.graph().opBuilder("Variable", "rv_stage1_out");

        	builder2.addInput(builder_init);

        	//OperationBuilder builder2 = bundle.graph().opBuilder("Variable", "rv_stage1_out");

        	//builder2.addInput(builder_init);

        	

        	//Tensor<Float> tensorVal = t;

        	//Output oValue = graph.opBuilder("Const", name).setAttr("dtype", tensorVal.dataType()).setAttr("value", tensorVal).build().output(0);

        	

        	//bundle.graph().opBuilder("Assign", "Assign/rv_stage1_out").setAttr("value", t).build();

        	

        	Operation oprahWinfrey = bundle.graph().operation("rv_stage1_out");

        	System.out.println(oprahWinfrey.toString());

        	System.out.println(oprahWinfrey.type());



        	//Operation oprahWinfrey2 = bundle.graph().operation("rv_stage1_out_nonexistant");

        	//System.out.println(oprahWinfrey2.toString());

        	//System.out.println(oprahWinfrey2.type());

        	

        	System.out.println("DONE");

        	

        	//Tensor<Float> t = toTensor2DFloat(rv_stage1_out, tensorsToClose);

        	//builder.setAttr("dtype", t.dataType()).setAttr("shape",t.shape()).build().output(0);
@@ -186,6 +207,7 @@ public class TensorflowExamplePlugin 
        	

        	float [] resultValues = (float[]) result.copyTo(new float[78408]);

        	

        	

        	System.out.println("DONE");

        	

        } catch (final IllegalStateException ise) {