如何在 tensorflow.js 中截断 coco ssd 模型？

Question

ML / Tensorflow 初学者。

我在尝试从 coco ssd 模型中获取其中一层作为 React 应用程序中的包导入时遇到问题。我正在按照 Pacman tensorflow.js example 重新训练模型。

  const modelPromise = cocoSsd.load();
  Promise.all([modelPromise])
    .then(cocoModel => {
      console.log(cocoModel[0]);

      var cocoModel = cocoModel[0].model;
      console.log(cocoModel);

      const layer = cocoModel.getLayer('conv_pw_13_relu');
      this.truncatedCocoModel = tf.model({inputs: cocoModel.inputs, outputs:
                                                              layer.output});
   })
   .catch(error => {
      console.error(error);
   });

在 const 层行中，我收到 'cocoModel.getLayer is not a function' 的错误消息。 Pacman 示例使用的是 mobilenet 模型，我猜它具有此功能。

我在这里有哪些选择？我使用浏览器控制台环顾四周，但我无法在任何地方找到此功能，在线查找也没有太大帮助（是否有任何在线位置可以通过 Google 查看 cocoSSD 模型的整个结构？）

Answer 1

使用 npm 包 https://cdn.jsdelivr.net/npm/@tensorflow-models/coco-ssd，您无法检索任何图层。

load returns ObjectDetection 的一个实例，它没有 getLayer 属性.

如果要检索层，则必须按照所述加载图形模型

如何在 tensorflow.js 中截断 coco ssd 模型？

How to truncate coco ssd model in tensorflow.js?

javascript

machine-learning

tensorflow.js