如何在tensorflow.js中截断可可SSD模型？

Question

ML / Tensorflow初学者。

我在尝试从作为应用程序包中的包导入的coco ssd模型中获取一层时遇到了麻烦。 我正在遵循Pacman tensorflow.js示例来重新训练模型。

  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不是函数”。 Pacman示例使用的是mobilenet模型，我想它具有此功能。

我在这里有什么选择？ 我使用浏览器控制台环顾四周，但在任何地方都找不到此功能，并且在网上看也无济于事（在网上有什么地方可以看到Google的cocoSSD模型的整个结构吗？）

Answer 1

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

load返回一个不具有getLayer属性的ObjectDetection实例。

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