Tune scale size

hans00 · Oct 31, 2023 · f9e6a97 · f9e6a97
1 parent efc93fa
commit f9e6a97
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/components/models/ObjectDetection.tsx b/components/models/ObjectDetection.tsx
@@ -39,7 +39,7 @@ export function Interact({ settings: { model }, runPipe }: InteractProps): JSX.E
     setWIP(true);
     try {
       setInput(input);
-      const data = await getImageData(input);
+      const data = await getImageData(input, 128);
       const predicts = await runPipe('object-detection', model, createRawImage(data));
       setResults(predicts);
     } catch {}

diff --git a/utils/image.ts b/utils/image.ts
@@ -1,8 +1,6 @@
 import { RawImage } from '@xenova/transformers/src/utils/image';
 
-const maxSize = 768;
-
-export async function getImageData(uri: string): global.Image {
+export async function getImageData(uri: string, maxSize: number = 512): global.Image {
   return await new Promise((resolve, reject) => {
     const image = new Image();
     image.onload = () => {