java cnn图像识别_Java分布式神经网络库Deeplearning4j之上手实践手写数字图像识别与模型训练...

环境的搭建可以参考另一篇文章。

第一步运行MnistImagePipelineExampleSave代码下载数据集，并进行训练和保存

需要下载一个文件(windows默认保存在C:\Users\Administrator\AppData\Local\Temp\dl4j_Mnist)。文件存在git。如果网络不好。建议手动下载并解压。然后注释掉代码中的下载方法即可。如图所示:

训练需要一段时间等待即可。时间长短取决于自己电脑配置。

第二步运行MnistImagePipelineLoadChooser代码。并选中一个手写数字图像。进行识别测试

package org.deeplearning4j.examples.dataexamples;

import org.datavec.image.loader.NativeImageLoader;

import org.deeplearning4j.nn.multilayer.MultiLayerNetwork;

import org.deeplearning4j.util.ModelSerializer;

import org.nd4j.linalg.api.ndarray.INDArray;

import org.nd4j.linalg.data