Tensorflow.js tf.browser.toPixels()用法及代码示例

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

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

用法:

tf.browser.toPixels(img, Canvas);

参数：

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

返回值：它返回一个承诺，当渲染完成时，该承诺将解决。

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

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

输出：

tf.browser.toPixels() called

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

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