|
1 |
| -import WebAI from './src/webai.mjs' |
2 |
| -import cv from './src/opencv.mjs' |
| 1 | +import { WebAI, cv, ort } from './src/webai.mjs' |
3 | 2 | import YAML from 'yamljs'
|
4 | 3 |
|
| 4 | +window.WebAI = WebAI |
| 5 | +window.cv = cv |
| 6 | +window.ort = ort |
| 7 | + |
5 | 8 |
|
6 | 9 | // 获取 Doms
|
7 | 10 | const aDet = document.getElementById('aDet')
|
@@ -179,13 +182,13 @@ async function loadModel(MODE) {
|
179 | 182 | let modelURL = `${modelDir}/${MODE}/${modelName}/model.onnx.json`
|
180 | 183 | let modelConfig = `${modelDir}/${MODE}/${modelName}/configs.json`
|
181 | 184 | if (MODE == 'det') {
|
182 |
| - model = new WebAI.Det(modelURL, modelConfig, {executionProviders: ['wasm']}) |
| 185 | + model = new WebAI.Det(modelURL, modelConfig, { executionProviders: ['wasm'] }) |
183 | 186 | }
|
184 | 187 | else if (MODE == 'cls') {
|
185 |
| - model = new WebAI.Cls(modelURL, modelConfig, {executionProviders: ['webgpu']}) |
| 188 | + model = new WebAI.Cls(modelURL, modelConfig, { executionProviders: ['webgpu'] }) |
186 | 189 | }
|
187 | 190 | else if (MODE == 'seg') {
|
188 |
| - model = new WebAI.Seg(modelURL, modelConfig, {executionProviders: ['wasm']}) |
| 191 | + model = new WebAI.Seg(modelURL, modelConfig, { executionProviders: ['wasm'] }) |
189 | 192 | }
|
190 | 193 | inputFile.disabled = false
|
191 | 194 | buttonUser.disabled = false
|
|
0 commit comments