hehe

Author: chmjkb

apps/computer-vision/app/image_segmentation/index.tsx

@@ -89,7 +89,7 @@ export default function ImageSegmentationScreen() {
       setIsGenerating(true);
       const { width, height } = imageSize;
       const t1 = performance.now();
-      const output = await model.forward(imageUri, [], true);
+      const output = await model.forward(imageUri, ['PERSON'], true);
       const t2 = performance.now();
       console.log(t2 - t1);
       const argmax = output['ARGMAX'] || [];

packages/react-native-executorch/src/modules/computer_vision/ImageSegmentationModule.ts

@@ -102,14 +102,14 @@ export class ImageSegmentation<
     const normMean = [...(preprocessorConfig?.normMean ?? [])];
     const normStd = [...(preprocessorConfig?.normStd ?? [])];
     const paths = await ResourceFetcher.fetch(onDownloadProgress, modelSource);
-    if (paths === null || paths.length < 1) {
+    if (!paths?.[0]) {
       throw new RnExecutorchError(
         RnExecutorchErrorCode.DownloadInterrupted,
         'The download has been interrupted. Please retry.'
       );
     }
     const nativeModule = global.loadImageSegmentation(
-      paths[0] || '',
+      paths[0],
       normMean,
       normStd
     );
@@ -143,18 +143,12 @@ export class ImageSegmentation<
     onDownloadProgress: (progress: number) => void = () => {}
   ): Promise<ImageSegmentation<L>> {
     const paths = await ResourceFetcher.fetch(onDownloadProgress, modelSource);
-    if (paths === null || paths.length < 1) {
+    if (!paths?.[0]) {
       throw new RnExecutorchError(
         RnExecutorchErrorCode.DownloadInterrupted,
         'The download has been interrupted. Please retry.'
       );
     }
-    if (!paths[0]) {
-      throw new RnExecutorchError(
-        RnExecutorchErrorCode.DownloadInterrupted,
-        "The download couldn't be completed. Please retry."
-      );
-    }
     const normMean = config.preprocessorConfig?.normMean ?? [];
     const normStd = config.preprocessorConfig?.normStd ?? [];
     const nativeModule = global.loadImageSegmentation(