keras模型训练、保存与java调用

keras模型训练、保存与java调用

1. 模型的保存。因为我这里是用GPU训练的，但是想存成CPU模型。所以，我在原版代码里面添加了下面的设置。注意，只在save（）训练保存函数里面添加的。灵感来源添加设置的csdn链接

sess = tf.Session(config=tf.ConfigProto(device_count={'gpu':0}))

此外，模型的保存成.h5格式的。

    model.save('model\mnist.h5')

2.通过别人的模型转换脚本，将“.h5”模型转化为“.pb”模型。大神的转化代码链接如下：
代码转化脚本的github链接
3.之后就可以加载成功了
4.注意：
java中INDArray中元素只有4位小数点的精度，所以在tf中训练时，样本也需转换为4位的精度。比如:

x_train = np.around(x_train / 255.0, decimals=4)