为了账号安全,请及时绑定邮箱和手机立即绑定

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

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

慕尼黑8549860 2021-05-13 14:15:24
我已经按照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>我应该改变些什么来使其预测?
查看完整描述

1 回答

  • 1 回答
  • 0 关注
  • 407 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信