denoiser
Version:
OIDN Denoiser with tensorflow.js
15 lines (14 loc) • 854 B
TypeScript
import * as tf from '@tensorflow/tfjs';
export declare function getCorrectImageData(img: HTMLImageElement): ImageData;
export declare function hasSizeMissmatch(img: HTMLImageElement): boolean;
export declare function splitRGBA3D(inputTensor: tf.Tensor3D, disposeInputs?: boolean): Promise<{
rgb: tf.Tensor3D;
alpha: tf.Tensor3D;
}>;
export declare function concatenateAlpha3D(rgbTensor: tf.Tensor3D, alphaTensorIn?: tf.Tensor3D, disposeInputs?: boolean): tf.Tensor3D;
export declare function tensorLinearToSRGB(tensor: tf.Tensor3D): tf.Tensor3D;
export declare function tensorSRGBToLinear(tensor: tf.Tensor3D): tf.Tensor3D;
export declare function isMobile(): boolean;
export declare function logMemoryUsage(stage: string): void;
export declare function formatBytes(bytes: number): string;
export declare function formatTime(ms: number): string;