chore: dev

qq15725 · Dec 6, 2023 · d33c744 · d33c744
1 parent 9af84b1
commit d33c744
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 9 deletions.
diff --git 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
@@ -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
@@ -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()
 }