Class: ClassificationModule

Defined in: packages/react-native-executorch/src/modules/computer_vision/ClassificationModule.ts:12

Module for image classification tasks.

Extends

• BaseModule

Constructors

Constructor

new ClassificationModule(): ClassificationModule

Returns

ClassificationModule

Inherited from

BaseModule.constructor

Properties

nativeModule

nativeModule: any = null

Defined in: packages/react-native-executorch/src/modules/BaseModule.ts:8

Native module instance

Inherited from

BaseModule.nativeModule

Methods

delete()

delete(): void

Defined in: packages/react-native-executorch/src/modules/BaseModule.ts:41

Unloads the model from memory.

Returns

void

Inherited from

BaseModule.delete

---

forward()

forward(imageSource): Promise<{[category: string]: number; }>

Defined in: packages/react-native-executorch/src/modules/computer_vision/ClassificationModule.ts:43

Executes the model's forward pass, where imageSource can be a fetchable resource or a Base64-encoded string.

Parameters

imageSource

string

The image source to be classified.

Returns

Promise<{[category: string]: number; }>

The classification result.

---

forwardET()

protected forwardET(inputTensor): Promise<TensorPtr[]>

Defined in: packages/react-native-executorch/src/modules/BaseModule.ts:23

Runs the model's forward method with the given input tensors. It returns the output tensors that mimic the structure of output from ExecuTorch.

Parameters

inputTensor

TensorPtr[]

Array of input tensors.

Returns

Promise<TensorPtr[]>

Array of output tensors.

Inherited from

BaseModule.forwardET

---

getInputShape()

getInputShape(methodName, index): Promise<number[]>

Defined in: packages/react-native-executorch/src/modules/BaseModule.ts:34

Gets the input shape for a given method and index.

Parameters

methodName

string

method name

index

number

index of the argument which shape is requested

Returns

Promise<number[]>

The input shape as an array of numbers.

Inherited from

BaseModule.getInputShape

---

load()

load(model, onDownloadProgressCallback): Promise<void>

Defined in: packages/react-native-executorch/src/modules/computer_vision/ClassificationModule.ts:20

Loads the model, where modelSource is a string that specifies the location of the model binary. To track the download progress, supply a callback function onDownloadProgressCallback.

Parameters

model

Object containing modelSource.

modelSource

ResourceSource

onDownloadProgressCallback

(progress) => void

Optional callback to monitor download progress.

Returns

Promise<void>

Overrides

BaseModule.load