Как обрезать кокосовую модель ssd в tenorflow.js?

Question

ML / Tensorflow новичок.

У меня проблемы с попыткой получить один из слоев из модели coco ssd, импортированной как пакет в приложении React.Я следую примеру Pacman tenorflow.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);
   });

В строке константного слоя я получаю сообщение об ошибке, что «cocoModel.getLayer не является функцией»,Пример Pacman использует модель мобильной сети, которая, я думаю, имеет эту функцию

Какие у меня есть варианты?Я осмотрелся с помощью консоли браузера, но нигде не могу найти эту функцию, и поиск в Интернете мало помог (есть ли место в Интернете, где я могу увидеть всю структуру модели cocoSSD от Google?)

Answer 1

Используя пакет npm https://cdn.jsdelivr.net/npm/@tensorflow-models/coco-ssd,, вы не можете получить ни один слой.

load возвращает экземпляр ObjectDetection, у которого нет свойства getLayer.

Если вы хотите получить слой, вам нужно загрузить модель графика, как описано здесь