denoiser
Version:
OIDN Denoiser with tensorflow.js
15 lines (14 loc) • 499 B
TypeScript
import * as tf from '@tensorflow/tfjs';
export declare class GPUTensorTiler {
private model;
private tileSize;
private overlap;
private batchSize;
constructor(model: tf.LayersModel, tileSize: number, overlap?: number, batchSize?: number);
private getTileDims;
processLargeTensor(input: tf.Tensor4D): Promise<tf.Tensor4D>;
private createBlendingMask;
private reassembleTilesWithBlending;
dispose(): void;
}
export declare function testTiling(): Promise<void>;