本节涉及:
如果用户选择“y” ,则执行下面的步骤:
完后,会在程序执行目录下生成一个 export 子目录,其中包含了需要传递给其他语言程序的神经网络模型的相关文件。
在其他语言调用时,需要把这个文件夹 整个复制到需要使用的计算机上
调用模型文件进行预测的示例:
import org.tensorflow.Graph; import org.tensorflow.Session; import org.tensorflow.Tensor; import org.tensorflow.TensorFlow; import org.tensorflow.SavedModelBundle; import java.nio.FloatBuffer; import java.util.Arrays; public class TestTF { public static void main(String[] args) { SavedModelBundle smb = SavedModelBundle.load("export", "tag"); Session s = smb.session(); float[][] matrix = {{1.0F, 2.0F, 3.0F, 4.0F}}; System.out.println(Arrays.deepToString(matrix)); Tensor xFeed = Tensor.create(matrix); Tensor result = s.runner().feed("x", xFeed).fetch("y").run().get(0); FloatBuffer buf = FloatBuffer.allocate(2); result.writeTo(buf); System.out.println(result.toString()); System.out.println(buf.get(0)); System.out.println(buf.get(1)); } }
原文:https://www.cnblogs.com/expedition/p/11656593.html