问题标签 [tensorflowjs]

For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.

0 投票
1 回答
575 浏览

javascript - Keras 模型在转换为 tensorflow-js 模型后不提供相同的结果

Keras 模型在 python 中按预期执行,但在转换模型后,相同数据的结果不同。

我尝试更新 keras 和 tensorflow-js 版本,但仍然是同样的问题。

用于测试的 Python 代码:

结果:[[1.9591815e-16 1.0000000e+00 3.8602989e-18 3.2472009e-19 5.8910814e-11]] 1

这些结果是正确的。

Javascript代码:

tfjs 版本:

js中的preprocessing_img方法和预测:

在这种情况下,结果不一样,pred 数组中的最后一个索引是 1 90% 的时间。

我认为javascript中图像的预处理方法有问题,因为我不是javascript专家,还是我在javascript部分遗漏了什么?

0 投票
1 回答
785 浏览

javascript - Tensorflow 与 Tensorflow JS 浮点算术计算的不同结果

我已将 Tensorflow 模型转换为 Tensorflow JS 并尝试在浏览器中使用。在将输入输出输入模型进行推理之前,需要在输入输出上执行一些预处理步骤。我已经实现了与 Tensorflow 相同的这些步骤。问题是 TF JS 的推理结果与 Tensorflow 不同。所以我开始调试代码,发现TF JS预处理中浮点算术运算的结果与运行在带有GPU的Docker容器上的Tensorflow不同。TF JS 中使用的代码如下。

使用的 Python 代码块

规范化代码

调整大小图像代码

CopyChannels 代码

示例输出左侧是带有 GPU 的 Docker 上的 Tensorflow,右侧是 TF JS: 在此处输入图像描述

每一步之后的结果实际上是不同的。

0 投票
1 回答
554 浏览

javascript - 如何在不初始化权重的情况下导入训练模型

我使用 tensorflowjs-converter 将在 ImageNet 上预训练的 EfficientNet 模型转换为 tensorflow-js。当我尝试将模型加载到我的脚本中时,它会尝试使用未在 tfjs 中实现的初始化程序来初始化权重。然而,没有必要初始化权重,因为模型已经过预训练并且权重也被转换了。转换后的模型在这里: https ://github.com/paulsp94/tfjs_efficientnet3_imagenet

这是问题的 CodePen 示例: https ://codepen.io/paulsp94/pen/XLNdJq

您必须打开浏览器的控制台,才能看到正确的错误。

错误是:

错误:“未知初始化程序:EfficientConv2DKernelInitializer。这可能是由于以下原因之一:1.初始化程序是在 Python 中定义的,在这种情况下,它需要移植到 TensorFlow.js 或您的 JavaScript 代码。2.自定义初始化程序在 JavaScript 中定义,但未使用 tf.serialization.registerClass() 正确注册。”

更新: 虽然我可以绕过初始化程序,但将所有未知的初始化程序替换为例如 Zeros 初始化程序。我遇到了当前无法解决的自定义图层(Swish 图层)的另一个问题。

0 投票
1 回答
223 浏览

javascript - TensorFlow.js 是否使用 GPU 进行计算?

TensorFlow.js 的官方文档提到它使用 WebGL 来访问 GPU。WebGL 是否也仅用于渲染或计算?

0 投票
0 回答
189 浏览

javascript - tfjs-node-gpu 不使用 GPU

我正在尝试在使用节点包tfjs-node-gpu开发的 GPU 上运行 ML 代码。当我尝试这样做时,程序正在运行,但我看不到任何进程被添加到 GPU。由于这个原因,我怀疑我的程序没有使用 GPU。我刚刚安装了该软件包并在引用它的代码中使用。这是正确的方法吗我需要做更多的事情来让我的 TF node.js 代码在 GPU 上运行。

在我的代码开始执行之前,我看到了以下错误

在此处输入图像描述

以下是我的程序运行时的 GPU 日志

在此处输入图像描述

TIA

0 投票
1 回答
598 浏览

node.js - 如何仅使用 tensorflow 和 nodejs 制作自定义对象检测模型(cocossd)?

我需要一个关于如何创建对象检测的教程,使用 tensorflowjs 和 nodejs,没有浏览器,没有 python ......这可能吗,我试图开始这样做,但我得到了很多错误......所以我需要一个指南线来完成这一点。

我开始这样做

这让我遇到了这个错误。

UnhandledPromiseRejectionWarning: TypeError: tfconv.loadGraphModel 不是函数

0 投票
1 回答
215 浏览

tensorflowjs - 如何在 tensorflowjs 中克隆模型

在一组数据上训练一个模型后,我想得到它的一个副本并在不同的数据集上训练每个副本。有没有办法克隆一个模型,在克隆后可以分别训练每个模型?

我已经尝试将模型保存在我的本地存储中并从那里生成一个副本,但是 tensorflow 抱怨某些变量名称已经在使用中。

0 投票
1 回答
2275 浏览

javascript - 我可以在 tensorflowjs 中使用强化学习吗?

我目前正在做一个项目,该项目需要 tensorflowjs 创建一个从强化学习算法中学习的神经网络。这可能吗?

0 投票
1 回答
403 浏览

node.js - 无法在 node.js 中导入 @tensorflow-models

我刚刚开始研究 node.js,但无法将 tensorlfow js 模型导入我的应用程序。我正在使用的代码是:

运行上述代码后,我收到以下错误: