JPEG张量:提高质量

Question

我正在把从tensorflow相机返回的张量转换成jpeg。我已经做到了，但jpeg图像质量很差。这是为一个移动反应-本地应用开发与博览会。我用来将张量转换成图像的库是jpeg-js。

这是我用来将张量转换成图像的代码：

const handleCameraStream = async (imageAsTensors) => {
  const loop = async () => {
    const tensor = await imageAsTensors.next().value;
    console.log(tensor)
    const [height, width] = tensor.shape
    const data = new Buffer(
      // concat with an extra alpha channel and slice up to 4 channels to handle 3 and 4 channels tensors
      tf.concat([tensor, tf.ones([height, width, 1]).mul(255)], [-1])
        .slice([0], [height, width, 4])
        .dataSync(),
    )

    const rawImageData = {data, width, height};
    const jpegImageData = jpeg.encode(rawImageData, 200);

    const imgBase64 = tf.util.decodeString(jpegImageData.data, "base64")
    const salt = `${Date.now()}-${Math.floor(Math.random() * 10000)}`
    const uri = FileSystem.documentDirectory + `tensor-${salt}.jpg`;
    await FileSystem.writeAsStringAsync(uri, imgBase64, {
      encoding: FileSystem.EncodingType.Base64,
    });
    setUri(uri)
    // return {uri, width, height}
  };
  // requestAnimationFrameId = requestAnimationFrame(loop);
  !uri ? setTimeout(() => loop(), 2000) : null;
}

图像上半部分的图片是摄像机流。下面图片的下半部分是转换的张量。

​

​

Answer 1

我尝试了您的代码，增加了大小，高度和宽度，并能够获得一个良好的质量图像。如果这不起作用，也许你可以把剩下的代码发出来，我会尽力帮你的。

示例图像

    <TensorCamera
        type={Camera.Constants.Type.front}
        resizeHeight={cameraHeight}
        resizeWidth={cameraWidth}
        resizeDepth={3}
        onReady={handleCameraStream}
        autorender={true}
   />