react-native-executorch
Version:
An easy way to run AI models in React Native with ExecuTorch
12 lines (11 loc) • 537 B
TypeScript
import { ResourceSource } from '../../types/common';
import { DeeplabLabel } from '../../types/imageSegmentation';
import { BaseNonStaticModule } from '../BaseNonStaticModule';
export declare class ImageSegmentationModule extends BaseNonStaticModule {
load(model: {
modelSource: ResourceSource;
}, onDownloadProgressCallback?: (progress: number) => void): Promise<void>;
forward(imageSource: string, classesOfInterest?: DeeplabLabel[], resize?: boolean): Promise<{
[key in DeeplabLabel]?: number[];
}>;
}