ML/Tensorflow 初学者。
我在尝试从 React 应用程序中作为包导入的 coco ssd 模型中获取其中一层时遇到问题。我正在关注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 不是函数”。 Pacman 示例使用的是 mobilenet 模型,我猜它具有此功能。
我有什么选择?我使用浏览器控制台环顾四周,但在任何地方都找不到这个功能,网上查找也没有多大帮助(网上有什么地方可以看到 Google 的 cocoSSD 模型的整体结构吗?)
最佳答案
使用 npm 包 https://cdn.jsdelivr.net/npm/@tensorflow-models/coco-ssd ,您无法检索任何图层。
load返回没有 getLayer
属性的 ObjectDetection
实例。
如果您想检索图层,则必须按照 here 的描述加载图形模型。
关于javascript - 如何在tensorflow.js中 chop coco SSD模型?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54962038/