【问题标题】:TensorFlow Tensor to a TensorFlow.js tensorTensorFlow 张量到 TensorFlow.js 张量
【发布时间】:2019-07-05 09:04:09
【问题描述】:

我试图从 ConvNet 输出中获取张量并在 TF.js 中的另一个模型中使用它。有什么方法可以将 TF 张量保存到文件中并在 TF.js 中加载该文件并取回张量?

【问题讨论】:

  • 我已经加载了模型。但是输入是 TF python 中的张量,我想在 TF.js 模型中使用它。我需要以某种方式将 32x32x32x32 张量提取到 TF.js 张量

标签: javascript tensorflow keras deep-learning tensorflow.js


【解决方案1】:

张量可以保存到文件中。但是,要重新加载它需要一个服务器,因为浏览器不会直接访问文件系统,除非使用input type file

要将张量保存到文件中,必须先使用datadataSync 下载张量值。

const tensor = tf.tensor([1, 2])
// download values
const values = tensor.dataSync()

要查看如何将values 保存到文件中,可以使用以下answer

要加载保存的张量,文件内容需要由具有 http 请求的服务器提供服务。收到请求的响应后,

const tensorRetrieved = tf.tensor(contentHttpResponse) // if needed parse the string response using JSON.parse

另一种选择是在节点上使用 tensorflow.js。由于节点可以访问文件系统,因此不需要 http 请求和响应。

【讨论】:

  • 我得到了服务器,但是我需要一种方法将 TensorFlow 张量从 python 保存到本地磁盘,然后保存到服务器,然后从服务器保存到浏览器。但是我无法将 python 中的张量保存到本地磁盘...
猜你喜欢
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 2019-04-13
  • 2019-04-21
  • 1970-01-01
相关资源
最近更新 更多