TensorFlow.js中使用的Tensorflow对象检测API模型

问题描述 投票:0回答:1

我正在尝试使用在TensorFlow.js中使用TF对象检测API创建的对象检测模型。

我使用以下方法转换了模型:

tensorflowjs_converter --input_format = tf_saved_model推理图/ saved_model推理图/ web_model

它被转换而没有任何问题,并加载到我的JavaScript代码中。现在,我不确定下一步应该做什么。

使用python中的对象检测API,有很多步骤; (1)预处理图像,例如转换为RGB,改变numpy数组的形状,扩展尺寸(我已经知道如何处理)和(2)单个图像功能的运行推断,我不确定如何在tensorflow.js中进行处理。

[我只是尝试将图像放入模型中,并且有某种输出,但是看起来好像不是我想要的。

我知道该模型确实有效,正如我在Python中测试过的那样,这是我有点困惑的翻译成Javascript。

我试图找到有关在tensorflow.js中部署对象检测模型的一些常规信息,但是除了经过预训练的模型之外,我找不到太多其他信息。

有关此主题的任何信息都将非常有用!谢谢!

tensorflow object-detection-api tensorflow.js tensorflowjs-converter
1个回答
0
投票

您需要在javascript中复制相同的过程,然后再将其提供给模型。在js中,图片默认使用RGB通道,因此无需再次进行该转换。

© www.soinside.com 2019 - 2024. All rights reserved.