如何将我的会话或GAN模型保存到js文件中

时间:2019-07-02 16:40:37

标签: json tensorflow artificial-intelligence tensorflow.js

为此,我想在基于Web的UI上部署GANs模型,我需要将模型的检查点转换为js文件,以供网络代码调用。有用于save_model和keras的函数可以转换为pb文件,而对于js没有此功能

我主要担心的是我很困惑如何在js文件中转储会话或可变权重

1 个答案:

答案 0 :(得分:0)

您可以从python保存一个keras模型。有完整的教程here,但基本上等于在训练后称呼它:

driven_by :selenium, using: :selenium_chrome_billy

然后将结果托管在可公开访问的位置(或与Web UI放在同一服务器上),然后可以将模型加载到tensorflow.js中,如下所示:

tfjs.converters.save_keras_model(model, tfjs_target_dir)