SpringBoot整合TensorFlow_tensorflow springboot

简介

TensorFlow 是一个开源的、基于 Python 的机器学习框架，它由 Google 开发，并在图形分类、音频处理、推荐系统和自然语言处理等场景下有着丰富的应用，是目前最热门的机器学习框架，它可以帮助开发人员轻松获取数据，准备和训练模型，预测未来状态，以及执行大规模机器学习。除了 Python，TensorFlow 也提供了 C/C++、Java、Go、R 等其它编程语言的接口。但现在Java API几乎没有什么功能，TensorFlow唯一完全支持的语言是Python，所以就目前而言，可以用Python训练模型，而后投入生产时可以用Java调用模型，这篇文章主要是讲如何用SpringBoot简单整合Tensorflow!

pom.xml

在pom.xml中添加如下依赖

    <dependency>
          <groupId>org.tensorflow</groupId>
          <artifactId>tensorflow</artifactId>
          <version>1.15.0</version>
    </dependency>
1
2
3
4
5

如果不知道目前最新的，或者可以用的Tensorflow版本，可以去这里查看https://mvnrepository.com ，如果你的设备支持GPU功能，还可以添加以下依赖
注意：TensorFlow更新比较快，最好用哪个版本训练的模型，就用哪个版本调用！

    <dependency>
          <groupId>org.tensorflow</groupId>
          <artifactId>libtensorflow</artifactId>
          <version>1.15.0</version>
    </dependency>

    <dependency>
          <groupId>org.tensorflow</groupId>
          <artifactId>libtensorflow_jni_gpu</artifactId>
          <version>1.15.0</version>
    </dependency>
1
2
3
4
5
6
7
8
9
10
11

在项目包下随便新建一个包和类，例如我新建的是util/TestTensorflow.java，书写以下代码，我们来测试下，有没有整合成功

package com.example.tensorflow.util;

import org.tensorflow.TensorFlow;

/**
 * @ClassName TestTensorflow
 * @Description TODO
 * @Author L
 * @Date Create by 2020/2/21
 */
public class TestTensorflow {
    public static void main(String[] args) {
        System.out.println("TensorFlow version: "+TensorFlow.version());
    }
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

如果输出以下内容，说明整合成功了，对就是这么简单，一个依赖的事情

TensorFlow version: 1.15.0
1

最后，尽管Java API暂时还没有像TensorFlow在Python中那么多的功能，但是相信在不久的将来会有更好的支持，这算是一个良好的开端！