用 Java 实现深度学习 Demo [DL4J]

267 阅读2分钟

这是我参与11月更文挑战的第24天,活动详情查看:2021最后一次更文挑战

今天我们要说的主角是 Deeplearning4j 并且会给出如何从零搭建一个 Java 实现深度学习的 Demo。

今天的参考资料都在这里呢 deeplearning4j.konduit.ai/multi-proje…

image.png

什么是 Deeplearning4j

从目前的资料可以看出,dl4j 还处于初级结算,只有 1.0-M 版本,dl4j 开始作为一种特定域的语言来配置深度神经网络,并演变为一个可以供开发人员使用的工具,使用 Java 来实现模型训练并部署到生产环境。

前置条件

想要跑通 dl4j 的 demo 我们需要一点点的 Java 基础,比方说我们需要有 JVM 环境,这里要求要使用 JDK 8 及以上的版本。

我们可以使用 java -version 来检查一下本地的 JDK 版本,除此之外,还要对 maven 熟悉,然后需要做的就是 clone 项目,导入项目,就可以运行了,但是可惜的是这些 deep learning 的模型我都看不懂。今天写这个文章就算是一个科普吧,告诉大家一下,原来 Java 也可以跑深度学习了。而且还有专门的 jar 了,算是一种开眼吧。

想要运行 Deeplearning4j examples 还挺简单的,只要你会一点 Java 的知识配合 maven 的使用,还有就是网络要好,不然 jar 下载要一两个小时也是令人烦躁。

搭建 examples

只需三个命令,轻松搞定

git clone <https://github.com/eclipse/deeplearning4j-examples.git>

cd deeplearning4j-examples/dl4j-examples

mvn clean install

可能会出现什么样的坑呢?clone 没啥好说的,第二步中要注意,这个 dl4j-examples 是在根目录的里面的,要先进入项目根目录才行哈。

第三步就是靠网络取胜了,本地下载依赖到最后的测试打包,如果网络给力,一切正常那就美滋滋了。这里有一个技巧,就是 maven 配置一下阿里的镜像速度会快一点。

本来到这里就算结束了,但是奈何最后执行还是需要导入到 IDE 中的,这里以 IDEA 举例,我们需要配置好 IDEA 的 JDK、maven 然后就没有什么要做的了,除了等待项目加载完成,最后选择一个你感兴趣的特征或是模型,执行就好。

image-20211124221432759.png