在tensorflow.js中加载图像并转换为张量

// 获取用户选择的图像文件
const fileInput = document.getElementById('fileInput');
fileInput.addEventListener('change', (event) => {
  const file = event.target.files[0];

  // 创建一个新的img元素
  const img = new Image();

  // 监听图像加载完成事件
  img.onload = () => {
    // 创建一个新的canvas元素
    const canvas = document.createElement('canvas');
    const ctx = canvas.getContext('2d');

    // 将图像绘制到画布上
    ctx.drawImage(img, 0, 0);

    // 将画布中的图像转换为张量
    const tensor = tf.browser.fromPixels(canvas);

    // 可选的预处理操作
    const resizedTensor = tf.image.resizeBilinear(tensor, [224, 224]);
    const normalizedTensor = resizedTensor.div(255);

    // 打印张量的形状和数值
    console.log('Tensor shape:', normalizedTensor.shape);
    console.log('Tensor values:', normalizedTensor.dataSync());
  };

  // 读取图像文件的数据，并设置给img元素的src属性
  const reader = new FileReader();
  reader.onload = (event) => {
    img.src = event.target.result;
  };
  reader.readAsDataURL(file);
});