react-native-executorch/lib/module/modules/computer_vision/VisionLabeledModule.js

An easy way to run AI models in React Native with ExecuTorch

"use strict";

import { ResourceFetcher } from '../../utils/ResourceFetcher';
import { RnExecutorchErrorCode } from '../../errors/ErrorCodes';
import { RnExecutorchError } from '../../errors/errorUtils';
import { VisionModule } from './VisionModule';
export { ResolveLabels } from '../../types/computerVision';

/**
 * Fetches a model binary and returns its local path, throwing if the download
 * was interrupted (paused or cancelled).
 * @param source - The resource source to fetch the model binary from.
 * @param onDownloadProgress - Callback invoked with download progress (0–1).
 * @returns A promise resolving to the local file path of the downloaded model.
 * @internal
 */
export async function fetchModelPath(source, onDownloadProgress) {
  const paths = await ResourceFetcher.fetch(onDownloadProgress, source);
  if (!paths?.[0]) {
    throw new RnExecutorchError(RnExecutorchErrorCode.DownloadInterrupted, 'The download has been interrupted. Please retry.');
  }
  return paths[0];
}

/**
 * Base class for computer vision modules that carry a type-safe label map
 * and support the full VisionModule API (string/PixelData forward + runOnFrame).
 * @typeParam TOutput - The model's output type.
 * @typeParam LabelMap - The resolved {@link LabelEnum} for the model's output classes.
 * @internal
 */
export class VisionLabeledModule extends VisionModule {
  constructor(labelMap, nativeModule) {
    super();
    this.labelMap = labelMap;
    this.nativeModule = nativeModule;
  }
}
//# sourceMappingURL=VisionLabeledModule.js.map