無法在Tensorflow.js中加載凍結的模型

Question

我嘗試加載轉換后的Tensorflow模型，但控制台顯示以下錯誤：

版：

"dependencies": {
    "@tensorflow-models/posenet": "0.1.2",
    "@tensorflow/tfjs": "0.11.4",
    "@tensorflow/tfjs-converter": "^0.6.1",
    "asyncawait": "^1.0.7",
    "stats.js": "^0.17.0"
  }

碼：

import * as tf from '@tensorflow/tfjs';
import {loadFrozenModel} from '@tensorflow/tfjs-converter';

const MODEL_URL = 'faster_rcnn/web_model/tensorflowjs_model.pb';
const WEIGHTS_URL = 'faster_rcnn/web_model/weights_manifest.json';

const model =(async () => {
    await loadFrozenModel(MODEL_URL, WEIGHTS_URL)})();

錯誤：

我將模型放在Google雲端硬盤中：

https://drive.google.com/file/d/1yNIC9TD1MWzJyI3ewnJCmM1bmTAYC_1E/view?usp=sharing

請幫忙。

謝謝！

Answer 1

您必須將存儲類型傳遞給loadFrozenModel() 。 在您的情況下： file:/// 。

代碼中的另一個問題是，立即調用的異步函數不會返回任何內容，因此您的模型變量保持未定義狀態。

要解決這兩個問題：

(async () => {
    const model = loadFrozenModel('file:///' + MODEL_URL, 'file:///' + WEIGHTS_URL);
    //Now you can only use 'model' in here.
})();