keras2cpp

keras2cpp

这是一组代码,用于将 Keras 神经网络 模型移植到纯 C++ 中。神经网络 权重和架构存储在纯文本文件中,而图像的输入则以 vector >> 形式呈现。该代码已准备好支持简单的卷积网络(来自 MNIST 示例),但可以轻松扩展。仅实现了 ReLU 和 Softmax 激活函数。 它正在与 Theano 后端进行工作。

使用方法

1. 保存网络权重和架构。
2. 使用 dump_to_simple_cpp.py 脚本将网络结构储存到纯文本文件中。
3. 使用来自 keras_model.h 和 keras_model.cc 文件中的代码 - 请参阅下面的示例。

示例

1. 使用 example/mnist_cnn_one_iteration.py 脚本运行一次简单 CNN 迭代在 MNIST 数据上。它将生成具有架构 example/my_nn_arch.json 和以 HDF5 格式存储的权重 example/my_nn_weights.h5 的文件。

如果您想对网络进行测试，请使用 test_run.sh 脚本。请在其中提供您的网络架构和权重。该脚本执行以下工作：