Question

我正在尝试拍摄jpg照片，并对其进行灰度处理，然后将其调整为48x48的尺寸，我试过了，但是没用：

let image = require("../assets/angry.jpg");
      const imageAssetPath = Image.resolveAssetSource(image);
      const response = await fetch(imageAssetPath.uri, {}, { isBinary: true });
      const imageData = await response.arrayBuffer();
      let imageTensor = imageToTensor(imageData);
      const imageResize = tf.image.resizeBilinear(imageTensor, [48, 48], true);



  const imageToTensor = (rawData: ArrayBuffer) => {
    const { width, height, data } = jpeg.decode(rawData, true);
    const buffer = new Uint8Array(width * height * 3);
    let offset = 0;
    for (let i = 0; i < buffer.length; i += 3) {
      buffer[i] = data[offset]; //red
      buffer[i + 1] = data[offset + 1]; //green
      buffer[i + 2] = data[offset + 2]; //blue
      offset += 4; //skips Alpha value
    }
    return tf.tensor4d(buffer, [1, height, width, 3]);
  };

图像的大小调整为48x48，但是如何使其变为灰度？我尝试在imageToTensor函数中将数组更改为[height，width，1]，但它只会弄乱图片，有什么建议吗？

Answer 1

我不知道缺少这么多方法！您可以在python中查看rgb_to_grayscale的来源您将看到它们如何将rgb图像转换为灰度图像。

我试图以相同的方式在javascript中实现它，但是没有名为tf.tensordot的函数。

这是您的操作方式。

image = tf.ones([224, 224, 3])
rgb_weights = [0.2989, 0.5870, 0.1140]
image = tf.mul(image, rgb_weights)
image = tf.sum(image, axis=-1)
image = tf.expandDims(image, axis=-1)

调整大小并更改为灰度

1 个答案: