TensorFlow js 预测结果相同

Question

我通过将 python 模型转换为 tensorflow js 模型制作了一个 tensorflow js 模型。 python 模型预测图像并对其进行精细分类，但是使用 tensorflow js 模型，预测结果总是完全相同。

import * as tf from "@tensorflow/tfjs";
export default {
  data() {
    return {
      model: null,
      img: null,
      result: null,
    }
  },
  methods: {
    testModel() {
      const example = tf.browser.fromPixels(this.img)
      const prediction = this.model.predict(example.expandDims(0))
      this.result = prediction.argMax().dataSync()[0]
    }

  },
  async mounted() {
    this.model = await tf.loadLayersModel('/model.json')
    this.img = new Image()
    this.img.src = '/test_image.png'

  }
}

Answer 1

也许预测是在图像完成加载之前发生的。我认为图像将全是黑色像素。