image-classifier/lib/ImageClassifier.d.ts

Machine Learning Image Classifier for NodeJS

/// <reference types="node" />
declare class ImageClassifier {
    static default: typeof ImageClassifier;
    static create(): Promise<ImageClassifier>;
    static load(datasetPath: string): Promise<ImageClassifier>;
    private mobilenet;
    private classifier;
    private constructor();
    save(datasetDestination: string): Promise<void>;
    addExample(label: string, image: string | Buffer): Promise<void>;
    dropClassifier(label: string): void;
    predict(image: string | Buffer): Promise<{
        label: string;
        classIndex: number;
        confidences: {
            [label: string]: number;
        };
    }>;
}
export = ImageClassifier;
//# sourceMappingURL=ImageClassifier.d.ts.map