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

【问题标题】：Tensorflow Object Detection API model for use in TensorFlow.jsTensorFlow.js 中使用的 TensorFlow 对象检测 API 模型
【发布时间】：2020-03-02 23:14:32
【问题描述】：

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

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

tensorflowjs_converter --input_format=tf_saved_model inference_graph/saved_model inference_graph/web_model

它可以毫无问题地转换并加载到我的 javascript 代码中。现在我有点不确定我的下一步应该是什么。我必须将 Python 翻译成 JavaScript，但我不确定某些领域。

使用python中的对象检测API，步骤很多；（1）预处理图像，例如转换为 RGB、numpy 数组重塑、扩展尺寸（我知道如何处理它）和（2）单个图像函数的运行推断，我不确定我会如何在 tensorflow.js 中进行。

我试图找到一些有关在 tensorflow.js 中部署对象检测模型的一般信息，但除了预训练模型外，我找不到太多信息。

任何有关此主题的信息都会很棒！谢谢！

【问题讨论】：

标签： tensorflow object-detection-api tensorflow.js tensorflowjs-converter

【解决方案1】：

正如@edkeveked 所述，您需要在 JavaScript 中执行与在 Python 中类似的输入处理和输出处理。由于我不熟悉该模型，因此我无法确切说明您需要做什么。但是，您可以在此处找到使用特定对象检测模型的示例：

https://github.com/vabarbosa/tfjs-model-playground/blob/master/object-detector/demo/object-detector.js

另见

https://medium.com/codait/bring-machine-learning-to-the-browser-with-tensorflow-js-part-iii-62d2b09b10a3

【讨论】：

感谢您的资源！我肯定需要处理步骤，但运行推理部分我完全空白。你的帖子肯定对我有帮助！

【解决方案2】：

在将其提供给模型之前，您需要在 javascript 中复制相同的过程。在js中，图片默认使用RGB通道，无需再次转换。

【讨论】：

哦，是的，我确实需要相同的处理，我只是不确定某些区域的翻译。下次我会让我的问题更清楚。我不知道 JS 中的默认值是 RGB，所以感谢您提供的信息！