Commit 03e238e2 authored by Oleg Dzhimiev's avatar Oleg Dzhimiev
Browse files

more testing

parent 682c12fe
Loading
Loading
Loading
Loading
+31 −42
Original line number Diff line number Diff line
@@ -53,7 +53,7 @@ public class TensorflowExamplePlugin

    public final static String EXPORTDIR = "/home/oleg/GIT/python3-imagej-tiff/data_sets/tf_data_5x5_main_13_heur/exportdir";
    // tf.saved_model.tag_constants.SERVING = "serve"
    public final static String PB_TAG = "serve";
    public final static String SERVING = "serve";
    

    public static void run()
@@ -118,39 +118,20 @@ public class TensorflowExamplePlugin

        final Graph smpb;

        // init for variable?
        float [][] rv_stage1_out = new float[78408][32];
        
        // from: infer_qcds_01.py
        // from infer_qcds_01.py
        float [][] img_corr2d = new float[78408][324];
        float [][] img_target = new float[78408][  1];
        int     [] img_ntile  = new   int[78408];
        
        // init ntile
        // init ntile for testing?
        for(int i=0;i<img_ntile.length;i++){
        	img_ntile[i] = i;
        }
        
        /*
         * for feed:
         *     "ph_corr2d":           img_corr2d
         *     "ph_target_disparity": img_target
         *     "ph_ntile":            img_ntile
         *     
         *     so it will look like:
         *     
         *     https://divis.io/2018/01/enterprise-tensorflow-code-examples/ -> 
         *         https://github.com/DIVSIO/tensorflow_java_cli_example/blob/master/src/main/java/divisio/example/tensorflow/cli/RunRegression.java
         *     
         *     sess.runner()
         *         .feed("ph_corr2d",img_corr2d)
         *         .feed("ph_target_disparity",img_target)
         *         .feed("ph_ntile",img_ntile)
         *         .fetch("Disparity_net/stage1done:0")
         *         .run()
         *         .get(0)
         */
        
        final SavedModelBundle bundle = SavedModelBundle.load(EXPORTDIR,PB_TAG); 
        final SavedModelBundle bundle = SavedModelBundle.load(EXPORTDIR,SERVING); 
        
        final List<Tensor<?>> tensorsToClose = new ArrayList<Tensor<?>>(5);
        
@@ -158,31 +139,39 @@ public class TensorflowExamplePlugin
        
        try {
        	
        	//bundle.session().runner().fetch("Disparity_net/stage1done:0").run();
        	System.out.println("S0:");
        	// read Variable info test
        	Operation opr = bundle.graph().operation("rv_stage1_out");
        	System.out.println(opr.toString());
        	
        	System.out.println("S1:");
        	
        	// 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);
        	// init variable via constant?
        	Tensor<Float> tsr = toTensor2DFloat(rv_stage1_out, tensorsToClose);
        	Output builder_init = bundle.graph()
        			                    .opBuilder("Const", "rv_stage1_out_init")
        			                    .setAttr  ("dtype", tsr.dataType())
        			                    .setAttr  ("value", tsr)
        			                    .build()
        			                    .output(0);
        	
        	// variable
        	OperationBuilder builder2 = bundle.graph().opBuilder("Variable", "rv_stage1_out");
        	builder2.addInput(builder_init);
        	System.out.println(builder_init);
        	
        	//Tensor<Float> tensorVal = t;
        	//Output oValue = graph.opBuilder("Const", "rv_stage1_out").setAttr("dtype", tensorVal.dataType()).setAttr("value", tensorVal).build().output(0);
        	// variable
        	OperationBuilder builder2 = bundle.graph().opBuilder("Variable", "rv_stage1_out_extra_variable");
        											  //.addInput(builder_init);
        	
        	//bundle.graph().opBuilder("Assign", "Assign/rv_stage1_out").setAttr("value", t).build();
        	//builder2.
        	//bundle.graph().opBuilder("Assign", "Assign/" + builder2.op().name()).addInput(variable).addInput(value).build().output(0);
        	
        	Operation ooyoo = bundle.graph().operation("rv_stage1_out");
        	System.out.println(ooyoo.toString());
        	System.out.println(ooyoo.type());
        	//Tensor<Float> tensorVal = tsr;
        	//Output oValue = bundle.graph().opBuilder("Const", "rv_stage1_out_2").setAttr("dtype", tensorVal.dataType()).setAttr("value", tensorVal).build().output(0);
        	//System.out.println(oValue);
        	//Output oValue = bundle.graph().opBuilder("Variable", "rv_stage1_out").setAttr("value", tensorVal).build().output(0);
        	//bundle.graph().opBuilder("Assign", "Assign/rv_stage1_out").setAttr("value", tsr).build();
        	        	
        	System.out.println("DONE");
        	        	
        	//Tensor<Float> t = toTensor2DFloat(rv_stage1_out, tensorsToClose);
        	//builder.setAttr("dtype", t.dataType()).setAttr("shape",t.shape()).build().output(0);
        	
        	// stage 1        	
        	bundle.session().runner()
	                  .feed("ph_corr2d",toTensor2DFloat(img_corr2d, tensorsToClose))