node-efficientnet
Version:
Implementation of efficientNet model in nodejs
22 lines (21 loc) • 946 B
TypeScript
/// <reference types="node" />
import * as tf from "@tensorflow/tfjs-node-gpu";
import { io } from "@tensorflow/tfjs-core";
import { EfficientNetLabelLanguage, EfficientNetLanguageProvider } from "./EfficientNetLanguageProvider";
import EfficientNetResult from "./EfficientNetResult";
interface EfficientNetModelInferenceOptions {
topK?: number;
}
export default class EfficientNetModel {
modelPath: string | io.IOHandler;
imageSize: number;
model: tf.GraphModel | undefined;
languageProvider: EfficientNetLanguageProvider;
constructor(modelPath: string | io.IOHandler, imageSize: number, local: EfficientNetLabelLanguage | undefined);
load(): Promise<void>;
private createTensor;
private cropAndResize;
private predict;
inference(imgPath: string | Buffer, options?: EfficientNetModelInferenceOptions, overrideLanguageProvider?: EfficientNetLanguageProvider): Promise<EfficientNetResult>;
}
export {};