Función Tensorflow.js tf.browser.toPixels()

Tensorflow.js es una biblioteca de código abierto desarrollada por Google para ejecutar modelos de aprendizaje automático y redes neuronales de aprendizaje profundo en el entorno del navegador o del Node.

La función tf.browser.toPixels() se usa para convertir un tensor en una imagen en el navegador.

Sintaxis:

tf.browser.toPixels(img, Canvas);

Parámetros:

• img (tf.Tensor2D|tf.Tensor3D|TypedArray|Array): un tensor de rango 2 con forma [alto, ancho] o un tensor de rango 3 con forma [alto, ancho, número de canales].
• Lienzo [opcional] (HTMLCanvasElement): El lienzo para dibujar.

Valor de retorno: Devuelve una promesa que se resuelve cuando se completa el renderizado.

Ejemplo 1: En este ejemplo, estamos creando un tensor y llamando a la función tf.browser.toPixels() con el tensor.

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

Producción:

tf.browser.toPixels() called

Ejemplo 2: En este ejemplo, estamos creando un tensor y tomando la referencia del lienzo, y llamando a la función tf.browser.toPixels() con el tensor y la referencia del lienzo.

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
  );
});

Producción:

Make sure we cleaned up 2

Referencia: https://js.tensorflow.org/api/latest/#browser.toPixels