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