问题标签 [tfjs-node]

问问题

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.

53 问题

0 投票

1 回答

63 浏览

tensorflow-lite - 使用来自 Tensorflow Lite Model Maker 的 Saved Model Format 会导致预测结果不佳

按照本教程，我使用 Tensorflow Model Maker 创建了一个简单的图像分类模型。我将导出格式从 tflite 更改为 Saved Model，因为我打算将它与 tfjs-node 一起使用，如此处所示。

在 python 中使用图像测试 tflite 模型会返回非常好的结果（所需类接近 95%）：

使用 Saved Model 格式的预测结果完全不同且错误：

当我将保存的模型与 tfjs-node 一起使用时，同样的问题。

1.) 你知道如何解决这个问题或者我可以在哪里看看吗？2.) 还有什么可能有帮助的：tensorflow 中是否有任何受支持的方式可以将 tflite 模型与 tfjs-node 一起使用？

我在许多其他事情中尝试过的事情：

在创建模型之前将图像调整为所需大小以消除图像处理中的差异
将 input_tensor 转换为 float32 没有任何区别 (tf.convert_to_tensor(input_tensor, dtype=tf.float32))

tensorflow-lite tfjs-node

0 投票

0 回答

21 浏览

react-native - TFJS React Native giving different predictions on Android vs iOS

I'm trying to build an app that uses yolov5 (basically object detection). the inference results are the same as python when i use my samsung galaxy fold 3. however when i start using it on my iphone, the results starts becoming very random. for the same image, i got the following:

Android

but on iOS its

I found a github issue that says to do tf.ENV.set('WEBGL_PACK', false) but it did not help. apparently it's a webgl issue(?)

a snippet of my prediction function

any help would be greatly appreciated!