如何从本地磁盘加载bodypix模型?

时间:2020-05-02 15:11:52

标签: tensorflow.js tfjs-node

我的应用无法始终连接互联网,如何从本地磁盘加载经过训练的bodypix? nodejs代码段

    const net = await bodyPix.load({
        architecture: 'MobileNetV1',
        outputStride: 16,
        multiplier: 0.75,
        quantBytes: 2,
    });

在README.md中我找到了

modelUrl-一个可选的字符串,用于指定模型的自定义URL。 这对于本地开发或无法访问GCP托管的模型的国家很有用。

能否请您举一个例子,说明本地模型的网址是什么? dest模型的内容是什么(BTW:如何下载经过训练的模型?)。非常感谢!

2 个答案:

答案 0 :(得分:0)

答案 1 :(得分:0)

要在本地加载 bodypix 模型,您需要下载并托管模型的 json.bin 文件,然后使用 modelUrl 选项访问它们。

例如:

const net = await bodyPix.load({
    architecture: 'MobileNetV1',
    outputStride: 16,
    multiplier: 0.75,
    quantBytes: 2,
    modelUrl: "/public/model-stride16.json"
});

传递给 modelUrl 的网址必须是有效网址。