UNPKG

@vladmandic/face-api

Version:

JavaScript module for Face Detection and Face Recognition Using Tensorflow/JS

12 lines 606 B
import * as tf from '@tensorflow/tfjs/dist/tf.es2017.js'; import { env } from '../env'; import { isTensor4D } from '../utils'; export async function imageTensorToCanvas(imgTensor, canvas) { const targetCanvas = canvas || env.getEnv().createCanvasElement(); const [height, width, numChannels] = imgTensor.shape.slice(isTensor4D(imgTensor) ? 1 : 0); const imgTensor3D = tf.tidy(() => imgTensor.as3D(height, width, numChannels).toInt()); await tf.browser.toPixels(imgTensor3D, targetCanvas); imgTensor3D.dispose(); return targetCanvas; } //# sourceMappingURL=imageTensorToCanvas.js.map