keras2cpp:这是一堆将Keras神经网络模型移植到纯C ++中的代码

时间:2024-02-25 07:18:26
【文件属性】:

文件名称:keras2cpp:这是一堆将Keras神经网络模型移植到纯C ++中的代码

文件大小:59KB

文件格式:ZIP

更新时间:2024-02-25 07:18:26

machine-learning neural-network keras MachinelearningC++

keras2cpp 这是一堆将Keras神经网络模型移植到纯C ++中的代码。 神经网络权重和体系结构存储在纯文本文件中,对于图像,输入表示为vector > > 。 该代码准备支持简单的卷积网络(以MNIST为例),但可以轻松扩展。 仅实现ReLU和Softmax激活。 它正在与Theano后端一起使用。 用法 节省您的网络权重和架构。 使用dump_to_simple_cpp.py脚本将网络结构转储到纯文本文件。 将网络与keras_model.h和keras_model.cc文件中的代码keras_model.h使用-请参见下面的示例。


【文件预览】:
keras2cpp-master
----test_run_cnn.py(3KB)
----dump_to_simple_cpp.py(3KB)
----LICENSE(1KB)
----keras_model.h(7KB)
----example_main.cc(820B)
----README.md(2KB)
----example()
--------my_nn_arch.json(2KB)
--------sample_mnist.dat(7KB)
--------mnist_cnn_one_iteration.py(3KB)
--------dumped.nnet(60KB)
--------my_nn_weights.h5(36KB)
----test_run_cnn.cc(938B)
----keras_model.cc(14KB)
----test_run.sh(1KB)
----test_compare.py(1KB)

网友评论