提供的形状错误，预期 (1200000) 和实际 (400000) 张量值不同

Question

有没有办法让输入图像值与预期值匹配？我知道的输入图像的宽度是 800px，高度是 500px，并且是彩色的（意思是 3 个通道），所以形状应该是 [800,500,3]。 但是，我收到以下错误：

错误：Error: Based on the provided shape, [800,500,3], the tensor should have 1200000 values but has 400000

有没有什么方法可以给图像赋值 1200000，或者可以将形状调整为 400000 但仍然保留 3 个通道的 800x500 尺寸？

代码：

var tf = require('@tensorflow/tfjs-node');
const fs = require(`fs`)
const Jimp = require(`jimp`)

const image = `./1-1.png`

const imageWidth = 800;
const imageHeight = 500;
const imageChannels = 3;

const getData = async function (path) {
  const data = [];
  const image = await Jimp.read(path);
  await image
      .scan(0, 0, imageWidth, imageHeight, (x, y, idx) => {
        let v = image.bitmap.data[idx + 0];
        data.push(v / 255);
      });
  return data;
};

const createImage = async (data) => {
  const imTen = tf.tensor(data, [imageWidth, imageHeight, 3]);
  const inTen = imTen.expandDims();
  return inTen;
}

const main = async () => {
  const model = await tf.loadLayersModel('file:///retake/savedmodels/model.json');
  model.summary();

  const a = await getData(image)
  const b = await createImage(a)

  const tfImage = b
  console.log(im)

  const prediction = model.predict(tfImage);
  prediction.print();
}
main()

Answer 1

据我所知，输入图像的宽度为 800 像素，高度为 500 像素，并且是彩色的（意思是 3 个通道）

检查您的 getData 方法的返回值，因为它似乎返回的灰度图像 - 这将匹配 400,000 而不是预期的 1,200,000 值