无法通过在tensorflow.js中加载预训练的模型(loadLayersModel)进行预测

我已经按照TensorFlow.js自述文件中的说明训练并生成了文件


但是当我预测时,它是行不通的


<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs/dist/tf.min.js"></script>


<div>

  <h1 id="p">Try Tensorflow</h1>

  <p>model.json</p><input type="file" id="upload-json" />

  <p>weight.bin</p><input type="file" id="upload-weights" />

  <button type="button" id="myBtn" onclick="myFunction()">Try it</button>

  <script>

    function myFunction() {

      const uploadJSONInput = document.getElementById('upload-json');

      const uploadWeightsInput = document.getElementById('upload-weights');

      console.log('start');

      tf.tensor([

        [1, 2],

        [3, 4]

      ]).print(); //no issues umtill here


      const model = tf.loadLayersModel(tf.io.browserFiles(

        [uploadJSONInput.files[0], uploadWeightsInput.files[0]]

      )).then(() => {

        console.log('will print now');

        model.predict(tf.tensor2d([5], [1, 1])).print();

      });

      console.log(model.predict(tf.tensor2d([5], [1, 1])).print());


    }

  </script>

</div>

我应该改变些什么来使其预测?



慕尼黑8549860
浏览 408回答 1
1回答
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

JavaScript