import * as tf from "@tensorflow/tfjs";
import { NSFW_CLASSES } from "./nsfw_classes";
declare global {
    namespace NodeJS {
        interface Global {
            [x: string]: any;
        }
    }
    interface Window {
        [x: string]: any;
    }
}
type IOHandler = tf.io.IOHandler;
type ModelJSON = tf.io.ModelJSON;
export type FrameResult = {
    index: number;
    totalFrames: number;
    predictions: Array<PredictionType>;
    image: HTMLCanvasElement | ImageData;
};
export type ClassifyConfig = {
    topk?: number;
    fps?: number;
    onFrame?: (result: FrameResult) => any;
};
export interface NSFWJSOptions {
    size?: number;
    type?: string;
}
export type PredictionType = {
    className: (typeof NSFW_CLASSES)[keyof typeof NSFW_CLASSES];
    probability: number;
};
export type ModelName = "MobileNetV2" | "MobileNetV2Mid" | "InceptionV3";
export type ModelDefinition = {
    name: ModelName;
    numOfWeightBundles: number;
    options?: NSFWJSOptions;
    modelJson: () => Promise<{
        default: ModelJSON;
    }>;
    weightBundles: Array<() => Promise<{
        default: string;
    }>>;
};
export interface LoadOptions extends NSFWJSOptions {
    modelDefinitions?: ModelDefinition[];
}
export declare function load(modelOrUrl: ModelName): Promise<NSFWJS>;
export declare function load(modelOrUrl: string, options?: LoadOptions): Promise<NSFWJS>;
export declare class NSFWJS {
    endpoints: string[];
    model: tf.LayersModel | tf.GraphModel;
    private options;
    private urlOrIOHandler;
    private intermediateModels;
    private normalizationOffset;
    private disposed;
    constructor(modelUrlOrIOHandler: string | IOHandler, options: NSFWJSOptions);
    load(): Promise<void>;
    infer(img: tf.Tensor3D | ImageData | HTMLImageElement | HTMLCanvasElement | HTMLVideoElement, endpoint?: string): tf.Tensor;
    classify(img: tf.Tensor3D | ImageData | HTMLImageElement | HTMLCanvasElement | HTMLVideoElement, topk?: number): Promise<Array<PredictionType>>;
    dispose(): void;
}
export {};