12 lines
593 B
JavaScript
12 lines
593 B
JavaScript
|
import * as tf from '@tensorflow/tfjs-core';
|
||
|
|
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
|