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