在Java语言中载入Tensorflow模型并进行预测计算

最新推荐文章于 2024-07-26 10:05:28 发布

RorZoro

最新推荐文章于 2024-07-26 10:05:28 发布

阅读量1.7k

点赞数

分类专栏： Python TensorFlow 文章标签： TensorFlow模型

本文链接：https://blog.csdn.net/RorZoro/article/details/95855156

版权

目前，Tensorflow的Java版本支持Windows、Mac OS、Linux、Android这几个操作系统。本次主要以Windows操作系统为列来介绍。**
在Windows操作系统中，如果要在Java语言中调用TensorFlow的模型，需要到TensorFlow官网的安装页面中下载一个TensorFlow的工具类包libtensorflow-1.5.0.jar，还有一个包含JNI接口的动态链接库文件压缩包libtensorflow_jni-cpu-windows-x86_64-1.5.0.zip,该压缩包展开后会得到TensorFlow_jni.dll动态链接库文件、注意，文件名中的版本号部分可能随着TensorFlow的升级而有所变化，在使用java程序调用神经网络模型的时候，这里文件都会用到。
下面是调用代码保存的模型文件来进行预测的示例代码。

                             TestTF.java

import org.tensorflow.Graph;
import org.tensorflow.Session;
import org.tensorflow.Tensor;
import org.tensorflow.TensorFlow;
import org.tensorflow.SaveModelBundle;
import java.nio.FloatBuffer;
import java.util.Arrays;

public class TestTf {

public static void main(String[] args) {
SaveModelBudle smb =SaveModelBundle.load("export", "tag");
Session s=smb.session();

float[][] matrix={
  {1.0F,2.0F,3.0F,4.0F}};
System.out.println(Ar