diff --git a/components/models/ObjectDetection.tsx b/components/models/ObjectDetection.tsx index 602b8d3..82d1ee1 100644 --- 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 index f95fea2..2c66998 100644 --- 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 = () => {