Question

我首先在TensorFlow中训练模型，然后使用tf.saved_model.simple_save保存了模型。我已成功转换为tensorflowjs，但无法在tensorflowjs中使用tf.loadFrozenModel加载。

我已经在GitHub中上传了文件。这是项目：https://github.com/yiifanLu/tensorflowjs-problem。您能帮我检查出什么问题吗？

如果使用 web_model_less 文件夹，则浏览器可以正确输出。但是，当我使用 web_model 文件夹时，它始终处于等待状态。这些文件夹之间的区别是参数的数量。在 web_model_less 文件夹中，我仅使用4个卷积层。在 web_model 文件夹中，我使用8个卷积层。

我不知道如何解决这个问题！

Answer 1

我能够在计算机上加载模型，但是显然web_model目录中的模型很大（80MB），但是加载应该可以。

您在测试页上有一个小错误，应该在第二次调用时调用console.timeEnd。加载模型失败时，您可以共享控制台输出吗？

async function fun(){
    console.log("111");
    console.time('Loading of model');   
    const model_less = await tf.loadFrozenModel(MODEL_URL, WEIGHTS_URL);
    console.log("222");
    console.timeEnd('Loading of model');
}

为什么我不能在TensorFlow.js中使用tf.loadFrozenModel加载TensorFlow模型

1 个答案: