From d33c744f709e000e3fd106e55b19c74ebccfb28c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=BD=A0=E6=89=8B=E6=8D=A7=E9=B2=9C=E8=8A=B1?=
 <157215725@qq.com>
Date: Wed, 6 Dec 2023 19:09:46 +0800
Subject: [PATCH] chore: dev

---
 index.html               | 17 ++++++++++-------
 package.json             |  2 +-
 src/remove-background.ts |  2 +-
 3 files changed, 12 insertions(+), 9 deletions(-)
diff --git a/index.html b/index.html
index 955f63d..fb2c7cc 100644
--- a/index.html
+++ b/index.html
@@ -5,19 +5,22 @@
   <title>Playground</title>
 </head>
 <body>
-  <div>Playground</div>
-
   <script type="module" async>
-    import { removeBackground } from './dist'
+    import { removeBackground } from './src'
+
+    const imageSource = '/examples/example.jpg'
 
-    const blob = await removeBackground('/docs/assets/example.jpg', {
+    const blob = await removeBackground(imageSource, {
       debug: true,
-      model: '/models/u2net.onnx',
-      resolution: 320,
+      model: '/models/small.onnx',
+      resolution: 1024,
+      wasmPaths: {
+        'ort-wasm-simd.wasm': './node_modules/onnxruntime-web/dist/ort-wasm-simd.wasm',
+      },
     })
 
     const img1 = new Image()
-    img1.src = '/docs/assets/example.jpg'
+    img1.src = imageSource
     img1.width = '100'
     document.body.append(img1)
 
diff --git a/package.json b/package.json
index 1d268cd..dd2cab1 100644
--- a/package.json
+++ b/package.json
@@ -45,7 +45,7 @@
     "dist"
   ],
   "scripts": {
-    "dev": "rollup -cw & vite",
+    "dev": "vite",
     "lint": "eslint src",
     "test": "vitest",
     "build": "rimraf dist && rollup -c",
diff --git a/src/remove-background.ts b/src/remove-background.ts
index 16441b4..44c92e4 100644
--- a/src/remove-background.ts
+++ b/src/remove-background.ts
@@ -33,6 +33,7 @@ export async function removeBackground(
     resized.toBchwImageTensor().toTensor(),
   ])
   model.release()
+  debug && consoleDebug('Compute inference completion', result)
 
   const stride = resolution * resolution
   switch (options.output ?? 'foreground') {
@@ -64,6 +65,5 @@ export async function removeBackground(
       break
   }
 
-  debug && consoleDebug('Compute inference completion')
   return await resized.resize(imageTensor.dims[1], imageTensor.dims[0]).toBlob()
 }
