如何在不同的平台/技术中使用经过训练的神经网络?

时间:2018-06-19 17:17:02

标签: python c++ tensorflow neural-network

鉴于我在笔记本电脑上使用Tensorflow和Python训练了一个简单的神经网络,所以我想在C ++应用程序的手机上使用此模型。

我可以使用任何兼容格式吗? 运行神经网络(不训练)的最小框架是什么?

UDP。我也对Tensorflow与NOT-Tensorflow的兼容性感兴趣。我需要从头开始构建它,还是有最佳实践?

2 个答案:

答案 0 :(得分:3)

是,如果您使用的是iOS或Android。根据您的特定需求,您可以在TensorFlow for Mobile和TensorFlow Lite之间进行选择

https://www.tensorflow.org/mobile/

尤其是要加载预训练的模型 https://www.tensorflow.org/mobile/prepare_models

答案 1 :(得分:1)

从技术上讲,您根本不需要框架。常规的全连接神经网络非常简单,您可以在直接的C ++中实现它。矩阵乘法大约需要100行代码,非线性激活函数大约需要12行代码。

最大的一部分是弄清楚如何解析序列化的Tensorflow模型,特别是考虑到有很多方法可以这样做。您可能想要冻结您的TensorFlow模型;这会将最新训练的权重插入模型。