Tensorflow.js tf.browser.toPixels()函数

Tensorflow.js 是谷歌开发的一个开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。

tf.browser.toPixels()函数用于将张量转换为浏览器中的图像。

句法：

tf.browser.toPixels(img, Canvas);

参数：

img (tf.Tensor2D|tf.Tensor3D|TypedArray|Array)：形状为 [height, width] 的 rank-2 张量，或形状为 [height, width, numChannels] 的 rank-3 张量。
Canvas [可选] (HTMLCanvasElement)：要绘制到的画布。

返回值：它返回一个在渲染完成时解析的承诺。

示例 1：在此示例中，我们正在创建一个张量并使用该张量调用 tf.browser.toPixels()函数。

Javascript

import * as tf from "@tensorflow/tfjs"
const tensorA = tf.randomUniform([400, 400, 3]);
 
tf.browser.toPixels(tensorA).then(() => {
    console.log("tf.browser.toPixels() called");
});

Javascript

const tensorA = tf.randomUniform([400, 400, 3]);
 
const canvasA = document.getElementById("CanvasHTML");
 
tf.browser.toPixels(tensorA, canvasA).then(() => {
  tensorA.dispose();
  console.log(
    "Make sure we cleaned up",
    tf.memory().numTensors
  );
});

输出：

tf.browser.toPixels() called

示例 2：在此示例中，我们正在创建一个张量并获取画布引用，并使用张量和画布引用调用 tf.browser.toPixels()函数。

Javascript

const tensorA = tf.randomUniform([400, 400, 3]);
 
const canvasA = document.getElementById("CanvasHTML");
 
tf.browser.toPixels(tensorA, canvasA).then(() => {
  tensorA.dispose();
  console.log(
    "Make sure we cleaned up",
    tf.memory().numTensors
  );
});

输出：

Make sure we cleaned up 2

参考： https://js.tensorflow.org/api/latest/#browser.toPixels