@kongyo2/kongyo2x 0.2.0 → 0.3.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +2 -2
- package/dist/engine/cpuConv.js +15 -1
- package/dist/engine/cpuConv.js.map +1 -1
- package/dist/engine/deconv.js +12 -1
- package/dist/engine/deconv.js.map +1 -1
- package/dist/engine/runModel.js +11 -0
- package/dist/engine/runModel.js.map +1 -1
- package/dist/image/alpha.js +14 -2
- package/dist/image/alpha.js.map +1 -1
- package/dist/image/resize.js +12 -1
- package/dist/image/resize.js.map +1 -1
- package/dist/train/conv.js +22 -1
- package/dist/train/conv.js.map +1 -1
- package/dist/wasm/loader.d.ts +11 -0
- package/dist/wasm/loader.js +209 -0
- package/dist/wasm/loader.js.map +1 -0
- package/package.json +8 -3
- package/wasm/kongyo2x.wasm +0 -0
package/README.md
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
# kongyo2x
|
|
2
2
|
|
|
3
|
-
Image super-resolution in TypeScript. MLPconv networks, trained from scratch and
|
|
3
|
+
Image super-resolution in TypeScript. MLPconv networks, trained from scratch, upscale and denoise images — with the heavy convolutions, deconvolution, resampling, and training running in a bundled Rust/WebAssembly core (and a pure-TypeScript fallback). No Python.
|
|
4
4
|
|
|
5
5
|
## Install
|
|
6
6
|
|
|
@@ -8,7 +8,7 @@ Image super-resolution in TypeScript. MLPconv networks, trained from scratch and
|
|
|
8
8
|
npm install @kongyo2/kongyo2x
|
|
9
9
|
```
|
|
10
10
|
|
|
11
|
-
Requires Node.js 20+.
|
|
11
|
+
Requires Node.js 20+. The heavy math runs in a prebuilt Rust/WebAssembly module that ships with the package — no toolchain or build step at install time, and a pure-TypeScript fallback covers any environment where the module can't load. GPU acceleration is still available: brain.js and its `gpu.js` peer compile a native module (`gl`, headless-gl) at install time, which needs X11/OpenGL system libraries and can't complete in a headless environment, so brain.js is an optional dependency — when its native build fails npm skips it and the install still succeeds.
|
|
12
12
|
|
|
13
13
|
## CLI
|
|
14
14
|
|
package/dist/engine/cpuConv.js
CHANGED
|
@@ -1,5 +1,6 @@
|
|
|
1
|
-
import { createTensor } from "../core/tensor.js";
|
|
1
|
+
import { createTensor, fromData } from "../core/tensor.js";
|
|
2
2
|
import { convOutputSize } from "./conv.js";
|
|
3
|
+
import { getWasm, disableWasm } from "../wasm/loader.js";
|
|
3
4
|
export function cpuConvForward(input, layer, weights, bias, alpha) {
|
|
4
5
|
if (input.channels !== layer.inputPlanes) {
|
|
5
6
|
throw new Error(`conv expected ${layer.inputPlanes} input planes, received ${input.channels}`);
|
|
@@ -9,6 +10,19 @@ export function cpuConvForward(input, layer, weights, bias, alpha) {
|
|
|
9
10
|
const inW = input.width;
|
|
10
11
|
const outH = convOutputSize(inH, kH, strideY, padY);
|
|
11
12
|
const outW = convOutputSize(inW, kW, strideX, padX);
|
|
13
|
+
if (outH <= 0 || outW <= 0) {
|
|
14
|
+
throw new Error(`conv kernel ${kW}x${kH} is larger than padded input ${inW}x${inH}`);
|
|
15
|
+
}
|
|
16
|
+
const wasm = getWasm();
|
|
17
|
+
if (wasm) {
|
|
18
|
+
try {
|
|
19
|
+
const data = wasm.convForward(input.data, inH, inW, weights, bias, inputPlanes, outputPlanes, kW, kH, strideX, strideY, padX, padY, alpha, outH, outW);
|
|
20
|
+
return fromData(outputPlanes, outH, outW, data);
|
|
21
|
+
}
|
|
22
|
+
catch {
|
|
23
|
+
disableWasm();
|
|
24
|
+
}
|
|
25
|
+
}
|
|
12
26
|
const out = createTensor(outputPlanes, outH, outW);
|
|
13
27
|
const outData = out.data;
|
|
14
28
|
const inData = input.data;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"cpuConv.js","sourceRoot":"","sources":["../../src/engine/cpuConv.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;
|
|
1
|
+
{"version":3,"file":"cpuConv.js","sourceRoot":"","sources":["../../src/engine/cpuConv.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAG3D,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAEzD,MAAM,UAAU,cAAc,CAC5B,KAAa,EACb,KAAgB,EAChB,OAAqB,EACrB,IAAkB,EAClB,KAAa;IAEb,IAAI,KAAK,CAAC,QAAQ,KAAK,KAAK,CAAC,WAAW,EAAE,CAAC;QACzC,MAAM,IAAI,KAAK,CAAC,iBAAiB,KAAK,CAAC,WAAW,2BAA2B,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;IACjG,CAAC;IACD,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,KAAK,CAAC;IAC7G,MAAM,GAAG,GAAG,KAAK,CAAC,MAAM,CAAC;IACzB,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,CAAC;IACxB,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;IACpD,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;IACpD,IAAI,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,eAAe,EAAE,IAAI,EAAE,gCAAgC,GAAG,IAAI,GAAG,EAAE,CAAC,CAAC;IACvF,CAAC;IAED,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;IACvB,IAAI,IAAI,EAAE,CAAC;QACT,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAC3B,KAAK,CAAC,IAAI,EACV,GAAG,EACH,GAAG,EACH,OAAO,EACP,IAAI,EACJ,WAAW,EACX,YAAY,EACZ,EAAE,EACF,EAAE,EACF,OAAO,EACP,OAAO,EACP,IAAI,EACJ,IAAI,EACJ,KAAK,EACL,IAAI,EACJ,IAAI,CACL,CAAC;YACF,OAAO,QAAQ,CAAC,YAAY,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QAClD,CAAC;QAAC,MAAM,CAAC;YACP,WAAW,EAAE,CAAC;QAChB,CAAC;IACH,CAAC;IAED,MAAM,GAAG,GAAG,YAAY,CAAC,YAAY,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACnD,MAAM,OAAO,GAAG,GAAG,CAAC,IAAI,CAAC;IACzB,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC;IAC1B,MAAM,YAAY,GAAG,WAAW,GAAG,EAAE,GAAG,EAAE,CAAC;IAC3C,MAAM,SAAS,GAAG,IAAI,GAAG,IAAI,CAAC;IAE9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,SAAS,GAAG,IAAI,CAAC,CAAC,CAAW,CAAC;QACpC,MAAM,KAAK,GAAG,CAAC,GAAG,YAAY,CAAC;QAC/B,MAAM,QAAQ,GAAG,CAAC,GAAG,SAAS,CAAC;QAC/B,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;YACjC,MAAM,KAAK,GAAG,EAAE,GAAG,OAAO,GAAG,IAAI,CAAC;YAClC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;gBACjC,MAAM,KAAK,GAAG,EAAE,GAAG,OAAO,GAAG,IAAI,CAAC;gBAClC,IAAI,GAAG,GAAG,SAAS,CAAC;gBACpB,IAAI,CAAC,GAAG,KAAK,CAAC;gBACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,EAAE,CAAC,EAAE,EAAE,CAAC;oBACrC,MAAM,SAAS,GAAG,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC;oBAChC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;wBAC/B,MAAM,EAAE,GAAG,KAAK,GAAG,EAAE,CAAC;wBACtB,MAAM,UAAU,GAAG,EAAE,IAAI,CAAC,IAAI,EAAE,GAAG,GAAG,CAAC;wBACvC,MAAM,OAAO,GAAG,SAAS,GAAG,EAAE,GAAG,GAAG,CAAC;wBACrC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;4BAC/B,MAAM,EAAE,GAAG,KAAK,GAAG,EAAE,CAAC;4BACtB,IAAI,UAAU,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,GAAG,GAAG,EAAE,CAAC;gCACtC,GAAG,IAAK,MAAM,CAAC,OAAO,GAAG,EAAE,CAAY,GAAI,OAAO,CAAC,CAAC,CAAY,CAAC;4BACnE,CAAC;4BACD,CAAC,EAAE,CAAC;wBACN,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,OAAO,CAAC,QAAQ,GAAG,EAAE,GAAG,IAAI,GAAG,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,GAAG,GAAG,CAAC,CAAC;YAClE,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC"}
|
package/dist/engine/deconv.js
CHANGED
|
@@ -1,4 +1,5 @@
|
|
|
1
|
-
import { createTensor } from "../core/tensor.js";
|
|
1
|
+
import { createTensor, fromData } from "../core/tensor.js";
|
|
2
|
+
import { getWasm, disableWasm } from "../wasm/loader.js";
|
|
2
3
|
export function deconvOutputSize(inputSize, kernel, stride, pad, adj) {
|
|
3
4
|
return (inputSize - 1) * stride - 2 * pad + kernel + adj;
|
|
4
5
|
}
|
|
@@ -11,6 +12,16 @@ export function spatialFullConvolution(input, layer) {
|
|
|
11
12
|
const inW = input.width;
|
|
12
13
|
const outH = deconvOutputSize(inH, kH, strideY, padY, adjY);
|
|
13
14
|
const outW = deconvOutputSize(inW, kW, strideX, padX, adjX);
|
|
15
|
+
const wasm = getWasm();
|
|
16
|
+
if (wasm) {
|
|
17
|
+
try {
|
|
18
|
+
const data = wasm.deconvForward(input.data, inH, inW, layer.weights, layer.bias, layer.inputPlanes, layer.outputPlanes, kW, kH, strideX, strideY, padX, padY, adjX, adjY, outH, outW);
|
|
19
|
+
return fromData(layer.outputPlanes, outH, outW, data);
|
|
20
|
+
}
|
|
21
|
+
catch {
|
|
22
|
+
disableWasm();
|
|
23
|
+
}
|
|
24
|
+
}
|
|
14
25
|
const out = createTensor(layer.outputPlanes, outH, outW);
|
|
15
26
|
const outData = out.data;
|
|
16
27
|
const inData = input.data;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"deconv.js","sourceRoot":"","sources":["../../src/engine/deconv.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;
|
|
1
|
+
{"version":3,"file":"deconv.js","sourceRoot":"","sources":["../../src/engine/deconv.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAG3D,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAEzD,MAAM,UAAU,gBAAgB,CAAC,SAAiB,EAAE,MAAc,EAAE,MAAc,EAAE,GAAW,EAAE,GAAW;IAC1G,OAAO,CAAC,SAAS,GAAG,CAAC,CAAC,GAAG,MAAM,GAAG,CAAC,GAAG,GAAG,GAAG,MAAM,GAAG,GAAG,CAAC;AAC3D,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,KAAa,EAAE,KAAkB;IACtE,IAAI,KAAK,CAAC,QAAQ,KAAK,KAAK,CAAC,WAAW,EAAE,CAAC;QACzC,MAAM,IAAI,KAAK,CAAC,mBAAmB,KAAK,CAAC,WAAW,2BAA2B,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;IACnG,CAAC;IACD,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC;IAC9F,MAAM,GAAG,GAAG,KAAK,CAAC,MAAM,CAAC;IACzB,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,CAAC;IACxB,MAAM,IAAI,GAAG,gBAAgB,CAAC,GAAG,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IAC5D,MAAM,IAAI,GAAG,gBAAgB,CAAC,GAAG,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IAE5D,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;IACvB,IAAI,IAAI,EAAE,CAAC;QACT,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAC7B,KAAK,CAAC,IAAI,EACV,GAAG,EACH,GAAG,EACH,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,WAAW,EACjB,KAAK,CAAC,YAAY,EAClB,EAAE,EACF,EAAE,EACF,OAAO,EACP,OAAO,EACP,IAAI,EACJ,IAAI,EACJ,IAAI,EACJ,IAAI,EACJ,IAAI,EACJ,IAAI,CACL,CAAC;YACF,OAAO,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QACxD,CAAC;QAAC,MAAM,CAAC;YACP,WAAW,EAAE,CAAC;QAChB,CAAC;IACH,CAAC;IAED,MAAM,GAAG,GAAG,YAAY,CAAC,KAAK,CAAC,YAAY,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACzD,MAAM,OAAO,GAAG,GAAG,CAAC,IAAI,CAAC;IACzB,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC;IAC1B,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;IAC9B,MAAM,SAAS,GAAG,IAAI,GAAG,IAAI,CAAC;IAE9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5C,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAW,CAAC;QAC1C,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC;IAC9D,CAAC;IAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3C,MAAM,WAAW,GAAG,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC;QAClC,MAAM,YAAY,GAAG,CAAC,GAAG,KAAK,CAAC,YAAY,GAAG,EAAE,GAAG,EAAE,CAAC;QACtD,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,GAAG,EAAE,EAAE,EAAE,EAAE,CAAC;YAChC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,GAAG,EAAE,EAAE,EAAE,EAAE,CAAC;gBAChC,MAAM,CAAC,GAAG,MAAM,CAAC,WAAW,GAAG,EAAE,GAAG,GAAG,GAAG,EAAE,CAAW,CAAC;gBACxD,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;oBACZ,SAAS;gBACX,CAAC;gBACD,MAAM,MAAM,GAAG,EAAE,GAAG,OAAO,GAAG,IAAI,CAAC;gBACnC,MAAM,MAAM,GAAG,EAAE,GAAG,OAAO,GAAG,IAAI,CAAC;gBACnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC5C,MAAM,UAAU,GAAG,YAAY,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;oBAC9C,MAAM,YAAY,GAAG,CAAC,GAAG,SAAS,CAAC;oBACnC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;wBAC/B,MAAM,EAAE,GAAG,MAAM,GAAG,EAAE,CAAC;wBACvB,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,IAAI,IAAI,EAAE,CAAC;4BACzB,SAAS;wBACX,CAAC;wBACD,MAAM,UAAU,GAAG,YAAY,GAAG,EAAE,GAAG,IAAI,CAAC;wBAC5C,MAAM,aAAa,GAAG,UAAU,GAAG,EAAE,GAAG,EAAE,CAAC;wBAC3C,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;4BAC/B,MAAM,EAAE,GAAG,MAAM,GAAG,EAAE,CAAC;4BACvB,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,GAAG,IAAI,EAAE,CAAC;gCACzB,MAAM,GAAG,GAAG,UAAU,GAAG,EAAE,CAAC;gCAC5B,OAAO,CAAC,GAAG,CAAC,GAAI,OAAO,CAAC,GAAG,CAAY,GAAI,OAAO,CAAC,aAAa,GAAG,EAAE,CAAY,GAAG,CAAC,CAAC;4BACxF,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC"}
|
package/dist/engine/runModel.js
CHANGED
|
@@ -2,8 +2,10 @@ import { spatialFullConvolution } from "./deconv.js";
|
|
|
2
2
|
import { brainConvForward } from "./brainConv.js";
|
|
3
3
|
import { cpuConvForward } from "./cpuConv.js";
|
|
4
4
|
import { gpuConvForward, isGpuAvailable, disableGpu } from "./gpuConv.js";
|
|
5
|
+
import { isWasmAvailable } from "../wasm/loader.js";
|
|
5
6
|
export function runModel(model, input) {
|
|
6
7
|
let useGpu = isGpuAvailable();
|
|
8
|
+
const useWasm = isWasmAvailable();
|
|
7
9
|
let current = input;
|
|
8
10
|
for (let index = 0; index < model.layers.length; index++) {
|
|
9
11
|
const layer = model.layers[index];
|
|
@@ -24,6 +26,15 @@ export function runModel(model, input) {
|
|
|
24
26
|
}
|
|
25
27
|
}
|
|
26
28
|
}
|
|
29
|
+
// The Wasm CPU convolution outruns brain.js's per-patch forward, so prefer
|
|
30
|
+
// it whenever the layer exposes raw parameters and Wasm is available.
|
|
31
|
+
if (useWasm) {
|
|
32
|
+
const params = model.convParams(index);
|
|
33
|
+
if (params) {
|
|
34
|
+
current = cpuConvForward(current, layer, params.weights, params.bias, params.alpha);
|
|
35
|
+
continue;
|
|
36
|
+
}
|
|
37
|
+
}
|
|
27
38
|
const net = model.convNetwork(index);
|
|
28
39
|
if (net) {
|
|
29
40
|
current = brainConvForward(current, layer, net);
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"runModel.js","sourceRoot":"","sources":["../../src/engine/runModel.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;
|
|
1
|
+
{"version":3,"file":"runModel.js","sourceRoot":"","sources":["../../src/engine/runModel.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1E,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEpD,MAAM,UAAU,QAAQ,CAAC,KAAoB,EAAE,KAAa;IAC1D,IAAI,MAAM,GAAG,cAAc,EAAE,CAAC;IAC9B,MAAM,OAAO,GAAG,eAAe,EAAE,CAAC;IAClC,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC;QACzD,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,KAAK,CAAe,CAAC;QAChD,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC5B,OAAO,GAAG,sBAAsB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YACjD,SAAS;QACX,CAAC;QACD,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,MAAM,GAAG,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACvC,IAAI,MAAM,EAAE,CAAC;gBACX,IAAI,CAAC;oBACH,OAAO,GAAG,cAAc,CAAC,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;oBACpF,SAAS;gBACX,CAAC;gBAAC,MAAM,CAAC;oBACP,UAAU,EAAE,CAAC;oBACb,MAAM,GAAG,KAAK,CAAC;gBACjB,CAAC;YACH,CAAC;QACH,CAAC;QACD,2EAA2E;QAC3E,sEAAsE;QACtE,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,MAAM,GAAG,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACvC,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO,GAAG,cAAc,CAAC,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;gBACpF,SAAS;YACX,CAAC;QACH,CAAC;QACD,MAAM,GAAG,GAAG,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,GAAG,EAAE,CAAC;YACR,OAAO,GAAG,gBAAgB,CAAC,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC;YAChD,SAAS;QACX,CAAC;QACD,MAAM,MAAM,GAAG,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,GAAG,cAAc,CAAC,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;YACpF,SAAS;QACX,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,cAAc,KAAK,6EAA6E,CAAC,CAAC;IACpH,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC"}
|
package/dist/image/alpha.js
CHANGED
|
@@ -1,4 +1,5 @@
|
|
|
1
|
-
import { cloneTensor, createTensor } from "../core/tensor.js";
|
|
1
|
+
import { cloneTensor, createTensor, fromData } from "../core/tensor.js";
|
|
2
|
+
import { getWasm, disableWasm } from "../wasm/loader.js";
|
|
2
3
|
function box3x3Sum(plane, height, width) {
|
|
3
4
|
const out = new Float32Array(height * width);
|
|
4
5
|
for (let y = 0; y < height; y++) {
|
|
@@ -24,6 +25,17 @@ export function makeBorder(rgb, alpha, offset) {
|
|
|
24
25
|
throw new Error("makeBorder expects a 3-channel RGB tensor");
|
|
25
26
|
}
|
|
26
27
|
const { height, width } = rgb;
|
|
28
|
+
const steps = Number.isFinite(offset) && offset > 0 ? Math.floor(offset) : 0;
|
|
29
|
+
const wasm = getWasm();
|
|
30
|
+
if (wasm) {
|
|
31
|
+
try {
|
|
32
|
+
const data = wasm.makeBorder(rgb.data, alpha.data, height, width, steps);
|
|
33
|
+
return fromData(3, height, width, data);
|
|
34
|
+
}
|
|
35
|
+
catch {
|
|
36
|
+
disableWasm();
|
|
37
|
+
}
|
|
38
|
+
}
|
|
27
39
|
const size = height * width;
|
|
28
40
|
const eps = 1e-7;
|
|
29
41
|
const result = cloneTensor(rgb);
|
|
@@ -39,7 +51,7 @@ export function makeBorder(rgb, alpha, offset) {
|
|
|
39
51
|
out[2 * size + p] = 0;
|
|
40
52
|
}
|
|
41
53
|
}
|
|
42
|
-
for (let step = 0; step <
|
|
54
|
+
for (let step = 0; step < steps; step++) {
|
|
43
55
|
const maskWeight = box3x3Sum(mask, height, width);
|
|
44
56
|
for (let ch = 0; ch < 3; ch++) {
|
|
45
57
|
const channelBase = ch * size;
|
package/dist/image/alpha.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"alpha.js","sourceRoot":"","sources":["../../src/image/alpha.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;
|
|
1
|
+
{"version":3,"file":"alpha.js","sourceRoot":"","sources":["../../src/image/alpha.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAExE,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAEzD,SAAS,SAAS,CAAC,KAAmB,EAAE,MAAc,EAAE,KAAa;IACnE,MAAM,GAAG,GAAG,IAAI,YAAY,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC;IAC7C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAChC,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7B,MAAM,EAAE,GAAG,CAAC,GAAG,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;QAC/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/B,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC7B,MAAM,EAAE,GAAG,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC;YAC7C,IAAI,GAAG,GAAG,CAAC,CAAC;YACZ,KAAK,IAAI,EAAE,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;gBACjC,MAAM,OAAO,GAAG,EAAE,GAAG,KAAK,CAAC;gBAC3B,KAAK,IAAI,EAAE,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;oBACjC,GAAG,IAAI,KAAK,CAAC,OAAO,GAAG,EAAE,CAAW,CAAC;gBACvC,CAAC;YACH,CAAC;YACD,GAAG,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC;QAC3B,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,GAAW,EAAE,KAAa,EAAE,MAAc;IACnE,IAAI,GAAG,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;IAC/D,CAAC;IACD,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,CAAC;IAC9B,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAE7E,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;IACvB,IAAI,IAAI,EAAE,CAAC;QACT,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;YACzE,OAAO,QAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;QAC1C,CAAC;QAAC,MAAM,CAAC;YACP,WAAW,EAAE,CAAC;QAChB,CAAC;IACH,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,GAAG,KAAK,CAAC;IAC5B,MAAM,GAAG,GAAG,IAAI,CAAC;IACjB,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC;IAChC,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC;IAExB,IAAI,IAAI,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;IAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9B,IAAI,CAAC,CAAC,CAAC,GAAI,KAAK,CAAC,IAAI,CAAC,CAAC,CAAY,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAClD,CAAC;IACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9B,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;YAClB,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YACX,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;YAClB,GAAG,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;IAED,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC;QACxC,MAAM,UAAU,GAAG,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;QAClD,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC;YAC9B,MAAM,WAAW,GAAG,EAAE,GAAG,IAAI,CAAC;YAC9B,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW,GAAG,IAAI,CAAiB,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;YACxG,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9B,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;oBAClB,GAAG,CAAC,WAAW,GAAG,CAAC,CAAC,GAAI,OAAO,CAAC,CAAC,CAAY,GAAG,CAAE,UAAU,CAAC,CAAC,CAAY,GAAG,GAAG,CAAC,CAAC;gBACpF,CAAC;YACH,CAAC;QACH,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;QACxC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9B,QAAQ,CAAC,CAAC,CAAC,GAAI,UAAU,CAAC,CAAC,CAAY,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACtD,CAAC;QACD,IAAI,GAAG,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACpC,MAAM,CAAC,GAAG,GAAG,CAAC,CAAC,CAAW,CAAC;QAC3B,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,KAAa;IAC5C,IAAI,KAAK,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;QACzD,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;QACnE,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}
|
package/dist/image/resize.js
CHANGED
|
@@ -1,4 +1,5 @@
|
|
|
1
|
-
import { createTensor } from "../core/tensor.js";
|
|
1
|
+
import { createTensor, fromData } from "../core/tensor.js";
|
|
2
|
+
import { getWasm, disableWasm } from "../wasm/loader.js";
|
|
2
3
|
export function resizeNearest(input, outW, outH) {
|
|
3
4
|
const inH = input.height;
|
|
4
5
|
const inW = input.width;
|
|
@@ -88,6 +89,16 @@ export function resizeLanczos(input, outW, outH, a = 3) {
|
|
|
88
89
|
const inW = input.width;
|
|
89
90
|
const xTap = buildLanczosTaps(inW, outW, a);
|
|
90
91
|
const yTap = buildLanczosTaps(inH, outH, a);
|
|
92
|
+
const wasm = getWasm();
|
|
93
|
+
if (wasm) {
|
|
94
|
+
try {
|
|
95
|
+
const data = wasm.resizeLanczos(input.data, input.channels, inH, inW, xTap.indices, xTap.weights, xTap.taps, outW, yTap.indices, yTap.weights, yTap.taps, outH);
|
|
96
|
+
return fromData(input.channels, outH, outW, data);
|
|
97
|
+
}
|
|
98
|
+
catch {
|
|
99
|
+
disableWasm();
|
|
100
|
+
}
|
|
101
|
+
}
|
|
91
102
|
const horizontal = createTensor(input.channels, inH, outW);
|
|
92
103
|
{
|
|
93
104
|
const src = input.data;
|
package/dist/image/resize.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"resize.js","sourceRoot":"","sources":["../../src/image/resize.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;
|
|
1
|
+
{"version":3,"file":"resize.js","sourceRoot":"","sources":["../../src/image/resize.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE3D,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAEzD,MAAM,UAAU,aAAa,CAAC,KAAa,EAAE,IAAY,EAAE,IAAY;IACrE,MAAM,GAAG,GAAG,KAAK,CAAC,MAAM,CAAC;IACzB,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,CAAC;IACxB,MAAM,GAAG,GAAG,YAAY,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACrD,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC;IACvB,MAAM,GAAG,GAAG,GAAG,CAAC,IAAI,CAAC;IACrB,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC;IAClC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;QACjC,IAAI,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;QAC/C,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;YACX,EAAE,GAAG,CAAC,CAAC;QACT,CAAC;aAAM,IAAI,EAAE,IAAI,GAAG,EAAE,CAAC;YACrB,EAAE,GAAG,GAAG,GAAG,CAAC,CAAC;QACf,CAAC;QACD,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC;IAChB,CAAC;IACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;QACxC,MAAM,QAAQ,GAAG,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC;QAC/B,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;QACjC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;YACjC,IAAI,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;YAC/C,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;gBACX,EAAE,GAAG,CAAC,CAAC;YACT,CAAC;iBAAM,IAAI,EAAE,IAAI,GAAG,EAAE,CAAC;gBACrB,EAAE,GAAG,GAAG,GAAG,CAAC,CAAC;YACf,CAAC;YACD,MAAM,MAAM,GAAG,QAAQ,GAAG,EAAE,GAAG,GAAG,CAAC;YACnC,MAAM,MAAM,GAAG,QAAQ,GAAG,EAAE,GAAG,IAAI,CAAC;YACpC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;gBACjC,GAAG,CAAC,MAAM,GAAG,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,GAAI,IAAI,CAAC,EAAE,CAAY,CAAW,CAAC;YAClE,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAQD,SAAS,IAAI,CAAC,CAAS;IACrB,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QACZ,OAAO,CAAC,CAAC;IACX,CAAC;IACD,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC;IACvB,OAAO,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC;AAC3B,CAAC;AAED,SAAS,aAAa,CAAC,CAAS,EAAE,CAAS;IACzC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACtB,OAAO,CAAC,CAAC;IACX,CAAC;IACD,OAAO,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;AAC/B,CAAC;AAED,SAAS,gBAAgB,CAAC,MAAc,EAAE,OAAe,EAAE,CAAS;IAClE,MAAM,KAAK,GAAG,MAAM,GAAG,OAAO,CAAC;IAC/B,MAAM,OAAO,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1C,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;IACjD,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,OAAO,GAAG,aAAa,CAAC,CAAC;IACxD,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,OAAO,GAAG,aAAa,CAAC,CAAC;IAC1D,MAAM,WAAW,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;QACjC,MAAM,MAAM,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,GAAG,KAAK,GAAG,GAAG,CAAC;QACvC,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,CAAC;QAC1C,IAAI,GAAG,GAAG,CAAC,CAAC;QACZ,MAAM,IAAI,GAAG,CAAC,GAAG,aAAa,CAAC;QAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;YACvC,MAAM,GAAG,GAAG,KAAK,GAAG,CAAC,CAAC;YACtB,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,WAAW,EAAE,CAAC,CAAC,CAAC;YACzD,IAAI,OAAO,GAAG,GAAG,CAAC;YAClB,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;gBAChB,OAAO,GAAG,CAAC,CAAC;YACd,CAAC;iBAAM,IAAI,OAAO,IAAI,MAAM,EAAE,CAAC;gBAC7B,OAAO,GAAG,MAAM,GAAG,CAAC,CAAC;YACvB,CAAC;YACD,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC;YAC5B,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;YACtB,GAAG,IAAI,CAAC,CAAC;QACX,CAAC;QACD,IAAI,GAAG,KAAK,CAAC,EAAE,CAAC;YACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvC,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,GAAI,OAAO,CAAC,IAAI,GAAG,CAAC,CAAY,GAAG,GAAG,CAAC;YAC1D,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;AACnD,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,KAAa,EAAE,IAAY,EAAE,IAAY,EAAE,CAAC,GAAG,CAAC;IAC5E,MAAM,GAAG,GAAG,KAAK,CAAC,MAAM,CAAC;IACzB,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,CAAC;IACxB,MAAM,IAAI,GAAG,gBAAgB,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IAC5C,MAAM,IAAI,GAAG,gBAAgB,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IAE5C,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;IACvB,IAAI,IAAI,EAAE,CAAC;QACT,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAC7B,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,QAAQ,EACd,GAAG,EACH,GAAG,EACH,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,IAAI,EACT,IAAI,EACJ,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,IAAI,EACT,IAAI,CACL,CAAC;YACF,OAAO,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QACpD,CAAC;QAAC,MAAM,CAAC;YACP,WAAW,EAAE,CAAC;QAChB,CAAC;IACH,CAAC;IAED,MAAM,UAAU,GAAG,YAAY,CAAC,KAAK,CAAC,QAAQ,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;IAC3D,CAAC;QACC,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC;QACvB,MAAM,GAAG,GAAG,UAAU,CAAC,IAAI,CAAC;QAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,MAAM,QAAQ,GAAG,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC;YAC/B,MAAM,QAAQ,GAAG,CAAC,GAAG,GAAG,GAAG,IAAI,CAAC;YAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC7B,MAAM,MAAM,GAAG,QAAQ,GAAG,CAAC,GAAG,GAAG,CAAC;gBAClC,MAAM,MAAM,GAAG,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC;gBACnC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;oBACjC,MAAM,IAAI,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC;oBAC5B,IAAI,GAAG,GAAG,CAAC,CAAC;oBACZ,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;wBACnC,GAAG,IAAK,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,CAAY,GAAI,GAAG,CAAC,MAAM,GAAI,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,CAAY,CAAY,CAAC;oBAC3G,CAAC;oBACD,GAAG,CAAC,MAAM,GAAG,EAAE,CAAC,GAAG,GAAG,CAAC;gBACzB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,GAAG,GAAG,YAAY,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACrD,CAAC;QACC,MAAM,GAAG,GAAG,UAAU,CAAC,IAAI,CAAC;QAC5B,MAAM,GAAG,GAAG,GAAG,CAAC,IAAI,CAAC;QACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,MAAM,QAAQ,GAAG,CAAC,GAAG,GAAG,GAAG,IAAI,CAAC;YAChC,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;YACjC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;gBACjC,MAAM,IAAI,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC;gBAC5B,MAAM,MAAM,GAAG,QAAQ,GAAG,EAAE,GAAG,IAAI,CAAC;gBACpC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;oBACjC,IAAI,GAAG,GAAG,CAAC,CAAC;oBACZ,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;wBACnC,MAAM,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,CAAW,CAAC;wBAC5C,GAAG,IAAK,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,CAAY,GAAI,GAAG,CAAC,QAAQ,GAAG,EAAE,GAAG,IAAI,GAAG,EAAE,CAAY,CAAC;oBACzF,CAAC;oBACD,GAAG,CAAC,MAAM,GAAG,EAAE,CAAC,GAAG,GAAG,CAAC;gBACzB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC"}
|
package/dist/train/conv.js
CHANGED
|
@@ -1,4 +1,5 @@
|
|
|
1
|
-
import { createTensor } from "../core/tensor.js";
|
|
1
|
+
import { createTensor, fromData } from "../core/tensor.js";
|
|
2
|
+
import { getWasm, disableWasm } from "../wasm/loader.js";
|
|
2
3
|
export function validOutputSize(inputSize, kernel) {
|
|
3
4
|
return inputSize - kernel + 1;
|
|
4
5
|
}
|
|
@@ -40,6 +41,16 @@ export function convForward(input, layer) {
|
|
|
40
41
|
if (outH <= 0 || outW <= 0) {
|
|
41
42
|
throw new Error(`conv kernel ${kw}x${kh} is larger than input ${input.width}x${input.height}`);
|
|
42
43
|
}
|
|
44
|
+
const wasm = getWasm();
|
|
45
|
+
if (wasm) {
|
|
46
|
+
try {
|
|
47
|
+
const data = wasm.convForwardTrain(input.data, input.height, input.width, layer.weights, layer.bias, cin, cout, kh, kw, outH, outW);
|
|
48
|
+
return fromData(cout, outH, outW, data);
|
|
49
|
+
}
|
|
50
|
+
catch {
|
|
51
|
+
disableWasm();
|
|
52
|
+
}
|
|
53
|
+
}
|
|
43
54
|
const k = cin * kh * kw;
|
|
44
55
|
const p = outH * outW;
|
|
45
56
|
const col = im2col(input, kh, kw, outH, outW, k);
|
|
@@ -68,6 +79,16 @@ export function convBackward(input, layer, dPre, computeGradInput) {
|
|
|
68
79
|
const inW = input.width;
|
|
69
80
|
const outH = dPre.height;
|
|
70
81
|
const outW = dPre.width;
|
|
82
|
+
const wasm = getWasm();
|
|
83
|
+
if (wasm) {
|
|
84
|
+
try {
|
|
85
|
+
const din = wasm.convBackwardTrain(input.data, inH, inW, layer.weights, dPre.data, outH, outW, layer.gradWeights, layer.gradBias, cin, cout, kh, kw, computeGradInput);
|
|
86
|
+
return din ? fromData(cin, inH, inW, din) : null;
|
|
87
|
+
}
|
|
88
|
+
catch {
|
|
89
|
+
disableWasm();
|
|
90
|
+
}
|
|
91
|
+
}
|
|
71
92
|
const k = cin * kh * kw;
|
|
72
93
|
const p = outH * outW;
|
|
73
94
|
const col = im2col(input, kh, kw, outH, outW, k);
|
package/dist/train/conv.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"conv.js","sourceRoot":"","sources":["../../src/train/conv.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;
|
|
1
|
+
{"version":3,"file":"conv.js","sourceRoot":"","sources":["../../src/train/conv.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC3D,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAczD,MAAM,UAAU,eAAe,CAAC,SAAiB,EAAE,MAAc;IAC/D,OAAO,SAAS,GAAG,MAAM,GAAG,CAAC,CAAC;AAChC,CAAC;AAED,IAAI,UAAU,GAAiB,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;AACnD,IAAI,WAAW,GAAiB,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;AAEpD,SAAS,MAAM,CAAC,MAAoB,EAAE,MAAc;IAClD,OAAO,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;AACrE,CAAC;AAED,SAAS,MAAM,CAAC,KAAa,EAAE,EAAU,EAAE,EAAU,EAAE,IAAY,EAAE,IAAY,EAAE,CAAS;IAC1F,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,CAAC;IACxB,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC;IACnC,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC;IAC1B,MAAM,GAAG,GAAG,KAAK,CAAC,QAAQ,CAAC;IAC3B,MAAM,GAAG,GAAG,MAAM,CAAC,UAAU,EAAE,IAAI,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC;IAChD,UAAU,GAAG,GAAG,CAAC;IACjB,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;QACjC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;YACjC,IAAI,GAAG,GAAG,CAAC,EAAE,GAAG,IAAI,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC;YAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC7B,MAAM,OAAO,GAAG,CAAC,GAAG,OAAO,CAAC;gBAC5B,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;oBAC/B,MAAM,KAAK,GAAG,OAAO,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC;oBAC7C,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;wBAC/B,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,MAAM,CAAC,KAAK,GAAG,EAAE,CAAW,CAAC;oBAC5C,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,KAAa,EAAE,KAAsB;IAC/D,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,KAAK,CAAC;IACpC,IAAI,KAAK,CAAC,QAAQ,KAAK,GAAG,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,iBAAiB,GAAG,2BAA2B,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;IACnF,CAAC;IACD,MAAM,IAAI,GAAG,eAAe,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IAC/C,MAAM,IAAI,GAAG,eAAe,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAC9C,IAAI,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,eAAe,EAAE,IAAI,EAAE,yBAAyB,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IACjG,CAAC;IAED,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;IACvB,IAAI,IAAI,EAAE,CAAC;QACT,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAChC,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,MAAM,EACZ,KAAK,CAAC,KAAK,EACX,KAAK,CAAC,OAAO,EACb,KAAK,CAAC,IAAI,EACV,GAAG,EACH,IAAI,EACJ,EAAE,EACF,EAAE,EACF,IAAI,EACJ,IAAI,CACL,CAAC;YACF,OAAO,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;QAC1C,CAAC;QAAC,MAAM,CAAC;YACP,WAAW,EAAE,CAAC;QAChB,CAAC;IACH,CAAC;IAED,MAAM,CAAC,GAAG,GAAG,GAAG,EAAE,GAAG,EAAE,CAAC;IACxB,MAAM,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;IACtB,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACjD,MAAM,GAAG,GAAG,YAAY,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IAC3C,MAAM,OAAO,GAAG,GAAG,CAAC,IAAI,CAAC;IACzB,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;IAC9B,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;IACxB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9B,MAAM,KAAK,GAAG,CAAC,GAAG,CAAC,CAAC;QACpB,MAAM,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC;QACtB,MAAM,SAAS,GAAG,IAAI,CAAC,CAAC,CAAW,CAAC;QACpC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC;YAC9B,MAAM,OAAO,GAAG,EAAE,GAAG,CAAC,CAAC;YACvB,IAAI,GAAG,GAAG,SAAS,CAAC;YACpB,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC;gBAC9B,GAAG,IAAK,OAAO,CAAC,KAAK,GAAG,EAAE,CAAY,GAAI,GAAG,CAAC,OAAO,GAAG,EAAE,CAAY,CAAC;YACzE,CAAC;YACD,OAAO,CAAC,OAAO,GAAG,EAAE,CAAC,GAAG,GAAG,CAAC;QAC9B,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,YAAY,CAC1B,KAAa,EACb,KAAsB,EACtB,IAAY,EACZ,gBAAyB;IAEzB,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,KAAK,CAAC;IACpC,MAAM,GAAG,GAAG,KAAK,CAAC,MAAM,CAAC;IACzB,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,CAAC;IACxB,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC;IACzB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC;IAExB,MAAM,IAAI,GAAG,OAAO,EAAE,CAAC;IACvB,IAAI,IAAI,EAAE,CAAC;QACT,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,CAAC,iBAAiB,CAChC,KAAK,CAAC,IAAI,EACV,GAAG,EACH,GAAG,EACH,KAAK,CAAC,OAAO,EACb,IAAI,CAAC,IAAI,EACT,IAAI,EACJ,IAAI,EACJ,KAAK,CAAC,WAAW,EACjB,KAAK,CAAC,QAAQ,EACd,GAAG,EACH,IAAI,EACJ,EAAE,EACF,EAAE,EACF,gBAAgB,CACjB,CAAC;YACF,OAAO,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACnD,CAAC;QAAC,MAAM,CAAC;YACP,WAAW,EAAE,CAAC;QAChB,CAAC;IACH,CAAC;IAED,MAAM,CAAC,GAAG,GAAG,GAAG,EAAE,GAAG,EAAE,CAAC;IACxB,MAAM,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;IACtB,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;IAC3B,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;IAC9B,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,CAAC;IACtC,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;IAEhC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9B,MAAM,KAAK,GAAG,CAAC,GAAG,CAAC,CAAC;QACpB,MAAM,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC;QACtB,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC;YAC9B,MAAM,CAAC,GAAG,QAAQ,CAAC,OAAO,GAAG,EAAE,CAAW,CAAC;YAC3C,OAAO,IAAI,CAAC,CAAC;YACb,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACZ,SAAS;YACX,CAAC;YACD,MAAM,OAAO,GAAG,EAAE,GAAG,CAAC,CAAC;YACvB,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC;gBAC9B,WAAW,CAAC,KAAK,GAAG,EAAE,CAAC,GAAI,WAAW,CAAC,KAAK,GAAG,EAAE,CAAY,GAAG,CAAC,GAAI,GAAG,CAAC,OAAO,GAAG,EAAE,CAAY,CAAC;YACpG,CAAC;QACH,CAAC;QACD,QAAQ,CAAC,CAAC,CAAC,GAAI,QAAQ,CAAC,CAAC,CAAY,GAAG,OAAO,CAAC;IAClD,CAAC;IAED,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IACxC,WAAW,GAAG,IAAI,CAAC;IACnB,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IACvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9B,MAAM,KAAK,GAAG,CAAC,GAAG,CAAC,CAAC;QACpB,MAAM,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC;QACtB,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC;YAC9B,MAAM,CAAC,GAAG,QAAQ,CAAC,OAAO,GAAG,EAAE,CAAW,CAAC;YAC3C,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACZ,SAAS;YACX,CAAC;YACD,MAAM,OAAO,GAAG,EAAE,GAAG,CAAC,CAAC;YACvB,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC;gBAC9B,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC,GAAI,IAAI,CAAC,OAAO,GAAG,EAAE,CAAY,GAAG,CAAC,GAAI,OAAO,CAAC,KAAK,GAAG,EAAE,CAAY,CAAC;YAC5F,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;IAC3C,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC;IAC5B,MAAM,OAAO,GAAG,GAAG,GAAG,GAAG,CAAC;IAC1B,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;QACjC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;YACjC,MAAM,OAAO,GAAG,CAAC,EAAE,GAAG,IAAI,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC;YACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC7B,MAAM,OAAO,GAAG,CAAC,GAAG,OAAO,CAAC;gBAC5B,MAAM,KAAK,GAAG,OAAO,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;gBACpC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;oBAC/B,MAAM,KAAK,GAAG,OAAO,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,GAAG,GAAG,GAAG,EAAE,CAAC;oBAC7C,MAAM,IAAI,GAAG,KAAK,GAAG,EAAE,GAAG,EAAE,CAAC;oBAC7B,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;wBAC/B,OAAO,CAAC,KAAK,GAAG,EAAE,CAAC,GAAI,OAAO,CAAC,KAAK,GAAG,EAAE,CAAY,GAAI,IAAI,CAAC,IAAI,GAAG,EAAE,CAAY,CAAC;oBACtF,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
export interface WasmKernels {
|
|
2
|
+
convForward(input: Float32Array, inH: number, inW: number, weights: Float32Array, bias: Float32Array, inputPlanes: number, outputPlanes: number, kW: number, kH: number, strideX: number, strideY: number, padX: number, padY: number, alpha: number, outH: number, outW: number): Float32Array;
|
|
3
|
+
deconvForward(input: Float32Array, inH: number, inW: number, weights: Float32Array, bias: Float32Array, inputPlanes: number, outputPlanes: number, kW: number, kH: number, strideX: number, strideY: number, padX: number, padY: number, adjX: number, adjY: number, outH: number, outW: number): Float32Array;
|
|
4
|
+
convForwardTrain(input: Float32Array, inH: number, inW: number, weights: Float32Array, bias: Float32Array, cin: number, cout: number, kh: number, kw: number, outH: number, outW: number): Float32Array;
|
|
5
|
+
convBackwardTrain(input: Float32Array, inH: number, inW: number, weights: Float32Array, dPre: Float32Array, outH: number, outW: number, gradWeights: Float32Array, gradBias: Float32Array, cin: number, cout: number, kh: number, kw: number, computeGradInput: boolean): Float32Array | null;
|
|
6
|
+
resizeLanczos(input: Float32Array, channels: number, inH: number, inW: number, xIdx: Int32Array, xWeights: Float32Array, xTaps: number, outW: number, yIdx: Int32Array, yWeights: Float32Array, yTaps: number, outH: number): Float32Array;
|
|
7
|
+
makeBorder(rgb: Float32Array, alpha: Float32Array, height: number, width: number, offset: number): Float32Array;
|
|
8
|
+
}
|
|
9
|
+
export declare function getWasm(): WasmKernels | undefined;
|
|
10
|
+
export declare function disableWasm(): void;
|
|
11
|
+
export declare function isWasmAvailable(): boolean;
|
|
@@ -0,0 +1,209 @@
|
|
|
1
|
+
import { readFileSync } from "node:fs";
|
|
2
|
+
const WASM_URL = new URL("../../wasm/kongyo2x.wasm", import.meta.url);
|
|
3
|
+
function isDisabled() {
|
|
4
|
+
return process.env.KONGYO2X_DISABLE_WASM === "1" || process.env.KONGYO2X_WASM === "0";
|
|
5
|
+
}
|
|
6
|
+
function instantiate() {
|
|
7
|
+
if (isDisabled()) {
|
|
8
|
+
return undefined;
|
|
9
|
+
}
|
|
10
|
+
try {
|
|
11
|
+
const bytes = readFileSync(WASM_URL);
|
|
12
|
+
const module = new WebAssembly.Module(bytes);
|
|
13
|
+
const instance = new WebAssembly.Instance(module, {});
|
|
14
|
+
const exports = instance.exports;
|
|
15
|
+
if (!(exports.memory instanceof WebAssembly.Memory)) {
|
|
16
|
+
return undefined;
|
|
17
|
+
}
|
|
18
|
+
const required = [
|
|
19
|
+
"kw_alloc",
|
|
20
|
+
"kw_dealloc",
|
|
21
|
+
"conv_forward",
|
|
22
|
+
"deconv_forward",
|
|
23
|
+
"conv_forward_train",
|
|
24
|
+
"conv_backward_train",
|
|
25
|
+
"resize_lanczos",
|
|
26
|
+
"make_border",
|
|
27
|
+
];
|
|
28
|
+
for (const name of required) {
|
|
29
|
+
if (typeof exports[name] !== "function") {
|
|
30
|
+
return undefined;
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
return exports;
|
|
34
|
+
}
|
|
35
|
+
catch {
|
|
36
|
+
return undefined;
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
function build(ex) {
|
|
40
|
+
const f32 = (ptr, len) => new Float32Array(ex.memory.buffer, ptr, len);
|
|
41
|
+
const i32 = (ptr, len) => new Int32Array(ex.memory.buffer, ptr, len);
|
|
42
|
+
const alloc = (elements) => ex.kw_alloc(elements * 4);
|
|
43
|
+
const free = (ptr, elements) => ex.kw_dealloc(ptr, elements * 4);
|
|
44
|
+
return {
|
|
45
|
+
convForward(input, inH, inW, weights, bias, ip, op, kW, kH, sx, sy, px, py, alpha, outH, outW) {
|
|
46
|
+
const inLen = ip * inH * inW;
|
|
47
|
+
const wLen = op * ip * kH * kW;
|
|
48
|
+
const outLen = op * outH * outW;
|
|
49
|
+
const inPtr = alloc(inLen);
|
|
50
|
+
const wPtr = alloc(wLen);
|
|
51
|
+
const bPtr = alloc(op);
|
|
52
|
+
const outPtr = alloc(outLen);
|
|
53
|
+
try {
|
|
54
|
+
f32(inPtr, inLen).set(input.subarray(0, inLen));
|
|
55
|
+
f32(wPtr, wLen).set(weights.subarray(0, wLen));
|
|
56
|
+
f32(bPtr, op).set(bias.subarray(0, op));
|
|
57
|
+
ex.conv_forward(inPtr, inH, inW, wPtr, bPtr, outPtr, ip, op, kW, kH, sx, sy, px, py, alpha);
|
|
58
|
+
return Float32Array.from(f32(outPtr, outLen));
|
|
59
|
+
}
|
|
60
|
+
finally {
|
|
61
|
+
free(inPtr, inLen);
|
|
62
|
+
free(wPtr, wLen);
|
|
63
|
+
free(bPtr, op);
|
|
64
|
+
free(outPtr, outLen);
|
|
65
|
+
}
|
|
66
|
+
},
|
|
67
|
+
deconvForward(input, inH, inW, weights, bias, ip, op, kW, kH, sx, sy, px, py, adjX, adjY, outH, outW) {
|
|
68
|
+
const inLen = ip * inH * inW;
|
|
69
|
+
const wLen = ip * op * kH * kW;
|
|
70
|
+
const outLen = op * outH * outW;
|
|
71
|
+
const inPtr = alloc(inLen);
|
|
72
|
+
const wPtr = alloc(wLen);
|
|
73
|
+
const bPtr = alloc(op);
|
|
74
|
+
const outPtr = alloc(outLen);
|
|
75
|
+
try {
|
|
76
|
+
f32(inPtr, inLen).set(input.subarray(0, inLen));
|
|
77
|
+
f32(wPtr, wLen).set(weights.subarray(0, wLen));
|
|
78
|
+
f32(bPtr, op).set(bias.subarray(0, op));
|
|
79
|
+
ex.deconv_forward(inPtr, inH, inW, wPtr, bPtr, outPtr, ip, op, kW, kH, sx, sy, px, py, adjX, adjY);
|
|
80
|
+
return Float32Array.from(f32(outPtr, outLen));
|
|
81
|
+
}
|
|
82
|
+
finally {
|
|
83
|
+
free(inPtr, inLen);
|
|
84
|
+
free(wPtr, wLen);
|
|
85
|
+
free(bPtr, op);
|
|
86
|
+
free(outPtr, outLen);
|
|
87
|
+
}
|
|
88
|
+
},
|
|
89
|
+
convForwardTrain(input, inH, inW, weights, bias, cin, cout, kh, kw, outH, outW) {
|
|
90
|
+
const inLen = cin * inH * inW;
|
|
91
|
+
const wLen = cout * cin * kh * kw;
|
|
92
|
+
const outLen = cout * outH * outW;
|
|
93
|
+
const inPtr = alloc(inLen);
|
|
94
|
+
const wPtr = alloc(wLen);
|
|
95
|
+
const bPtr = alloc(cout);
|
|
96
|
+
const outPtr = alloc(outLen);
|
|
97
|
+
try {
|
|
98
|
+
f32(inPtr, inLen).set(input.subarray(0, inLen));
|
|
99
|
+
f32(wPtr, wLen).set(weights.subarray(0, wLen));
|
|
100
|
+
f32(bPtr, cout).set(bias.subarray(0, cout));
|
|
101
|
+
ex.conv_forward_train(inPtr, inH, inW, wPtr, bPtr, outPtr, cin, cout, kh, kw);
|
|
102
|
+
return Float32Array.from(f32(outPtr, outLen));
|
|
103
|
+
}
|
|
104
|
+
finally {
|
|
105
|
+
free(inPtr, inLen);
|
|
106
|
+
free(wPtr, wLen);
|
|
107
|
+
free(bPtr, cout);
|
|
108
|
+
free(outPtr, outLen);
|
|
109
|
+
}
|
|
110
|
+
},
|
|
111
|
+
convBackwardTrain(input, inH, inW, weights, dPre, outH, outW, gradW, gradB, cin, cout, kh, kw, computeGradInput) {
|
|
112
|
+
const inLen = cin * inH * inW;
|
|
113
|
+
const wLen = cout * cin * kh * kw;
|
|
114
|
+
const dpreLen = cout * outH * outW;
|
|
115
|
+
const inPtr = alloc(inLen);
|
|
116
|
+
const wPtr = alloc(wLen);
|
|
117
|
+
const dprePtr = alloc(dpreLen);
|
|
118
|
+
const gradWPtr = alloc(wLen);
|
|
119
|
+
const gradBPtr = alloc(cout);
|
|
120
|
+
const dinPtr = computeGradInput ? alloc(inLen) : 0;
|
|
121
|
+
try {
|
|
122
|
+
f32(inPtr, inLen).set(input.subarray(0, inLen));
|
|
123
|
+
f32(wPtr, wLen).set(weights.subarray(0, wLen));
|
|
124
|
+
f32(dprePtr, dpreLen).set(dPre.subarray(0, dpreLen));
|
|
125
|
+
f32(gradWPtr, wLen).set(gradW.subarray(0, wLen));
|
|
126
|
+
f32(gradBPtr, cout).set(gradB.subarray(0, cout));
|
|
127
|
+
ex.conv_backward_train(inPtr, inH, inW, wPtr, dprePtr, outH, outW, gradWPtr, gradBPtr, dinPtr, cin, cout, kh, kw, computeGradInput ? 1 : 0);
|
|
128
|
+
gradW.set(f32(gradWPtr, wLen));
|
|
129
|
+
gradB.set(f32(gradBPtr, cout));
|
|
130
|
+
return computeGradInput ? Float32Array.from(f32(dinPtr, inLen)) : null;
|
|
131
|
+
}
|
|
132
|
+
finally {
|
|
133
|
+
free(inPtr, inLen);
|
|
134
|
+
free(wPtr, wLen);
|
|
135
|
+
free(dprePtr, dpreLen);
|
|
136
|
+
free(gradWPtr, wLen);
|
|
137
|
+
free(gradBPtr, cout);
|
|
138
|
+
if (dinPtr) {
|
|
139
|
+
free(dinPtr, inLen);
|
|
140
|
+
}
|
|
141
|
+
}
|
|
142
|
+
},
|
|
143
|
+
resizeLanczos(input, channels, inH, inW, xIdx, xWeights, xTaps, outW, yIdx, yWeights, yTaps, outH) {
|
|
144
|
+
const inLen = channels * inH * inW;
|
|
145
|
+
const xLen = outW * xTaps;
|
|
146
|
+
const yLen = outH * yTaps;
|
|
147
|
+
const outLen = channels * outH * outW;
|
|
148
|
+
const inPtr = alloc(inLen);
|
|
149
|
+
const xIdxPtr = alloc(xLen);
|
|
150
|
+
const xWPtr = alloc(xLen);
|
|
151
|
+
const yIdxPtr = alloc(yLen);
|
|
152
|
+
const yWPtr = alloc(yLen);
|
|
153
|
+
const outPtr = alloc(outLen);
|
|
154
|
+
try {
|
|
155
|
+
f32(inPtr, inLen).set(input.subarray(0, inLen));
|
|
156
|
+
i32(xIdxPtr, xLen).set(xIdx.subarray(0, xLen));
|
|
157
|
+
f32(xWPtr, xLen).set(xWeights.subarray(0, xLen));
|
|
158
|
+
i32(yIdxPtr, yLen).set(yIdx.subarray(0, yLen));
|
|
159
|
+
f32(yWPtr, yLen).set(yWeights.subarray(0, yLen));
|
|
160
|
+
ex.resize_lanczos(inPtr, channels, inH, inW, xIdxPtr, xWPtr, xTaps, outW, yIdxPtr, yWPtr, yTaps, outH, outPtr);
|
|
161
|
+
return Float32Array.from(f32(outPtr, outLen));
|
|
162
|
+
}
|
|
163
|
+
finally {
|
|
164
|
+
free(inPtr, inLen);
|
|
165
|
+
free(xIdxPtr, xLen);
|
|
166
|
+
free(xWPtr, xLen);
|
|
167
|
+
free(yIdxPtr, yLen);
|
|
168
|
+
free(yWPtr, yLen);
|
|
169
|
+
free(outPtr, outLen);
|
|
170
|
+
}
|
|
171
|
+
},
|
|
172
|
+
makeBorder(rgb, alpha, height, width, offset) {
|
|
173
|
+
const size = height * width;
|
|
174
|
+
const rgbLen = size * 3;
|
|
175
|
+
const rgbPtr = alloc(rgbLen);
|
|
176
|
+
const alphaPtr = alloc(size);
|
|
177
|
+
const outPtr = alloc(rgbLen);
|
|
178
|
+
try {
|
|
179
|
+
f32(rgbPtr, rgbLen).set(rgb.subarray(0, rgbLen));
|
|
180
|
+
f32(alphaPtr, size).set(alpha.subarray(0, size));
|
|
181
|
+
ex.make_border(rgbPtr, alphaPtr, outPtr, height, width, offset);
|
|
182
|
+
return Float32Array.from(f32(outPtr, rgbLen));
|
|
183
|
+
}
|
|
184
|
+
finally {
|
|
185
|
+
free(rgbPtr, rgbLen);
|
|
186
|
+
free(alphaPtr, size);
|
|
187
|
+
free(outPtr, rgbLen);
|
|
188
|
+
}
|
|
189
|
+
},
|
|
190
|
+
};
|
|
191
|
+
}
|
|
192
|
+
let resolved = false;
|
|
193
|
+
let kernels;
|
|
194
|
+
export function getWasm() {
|
|
195
|
+
if (!resolved) {
|
|
196
|
+
resolved = true;
|
|
197
|
+
const ex = instantiate();
|
|
198
|
+
kernels = ex ? build(ex) : undefined;
|
|
199
|
+
}
|
|
200
|
+
return kernels;
|
|
201
|
+
}
|
|
202
|
+
export function disableWasm() {
|
|
203
|
+
resolved = true;
|
|
204
|
+
kernels = undefined;
|
|
205
|
+
}
|
|
206
|
+
export function isWasmAvailable() {
|
|
207
|
+
return getWasm() !== undefined;
|
|
208
|
+
}
|
|
209
|
+
//# sourceMappingURL=loader.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"loader.js","sourceRoot":"","sources":["../../src/wasm/loader.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AA4KvC,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,0BAA0B,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAEtE,SAAS,UAAU;IACjB,OAAO,OAAO,CAAC,GAAG,CAAC,qBAAqB,KAAK,GAAG,IAAI,OAAO,CAAC,GAAG,CAAC,aAAa,KAAK,GAAG,CAAC;AACxF,CAAC;AAED,SAAS,WAAW;IAClB,IAAI,UAAU,EAAE,EAAE,CAAC;QACjB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC7C,MAAM,QAAQ,GAAG,IAAI,WAAW,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAkC,CAAC;QAC5D,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,YAAY,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC;YACpD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,QAAQ,GAAG;YACf,UAAU;YACV,YAAY;YACZ,cAAc;YACd,gBAAgB;YAChB,oBAAoB;YACpB,qBAAqB;YACrB,gBAAgB;YAChB,aAAa;SACd,CAAC;QACF,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;YAC5B,IAAI,OAAO,OAAO,CAAC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;gBACxC,OAAO,SAAS,CAAC;YACnB,CAAC;QACH,CAAC;QACD,OAAO,OAAiC,CAAC;IAC3C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,SAAS,KAAK,CAAC,EAAe;IAC5B,MAAM,GAAG,GAAG,CAAC,GAAW,EAAE,GAAW,EAAgB,EAAE,CAAC,IAAI,YAAY,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;IACrG,MAAM,GAAG,GAAG,CAAC,GAAW,EAAE,GAAW,EAAc,EAAE,CAAC,IAAI,UAAU,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;IACjG,MAAM,KAAK,GAAG,CAAC,QAAgB,EAAU,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC;IACtE,MAAM,IAAI,GAAG,CAAC,GAAW,EAAE,QAAgB,EAAQ,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,QAAQ,GAAG,CAAC,CAAC,CAAC;IAEvF,OAAO;QACL,WAAW,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI;YAC3F,MAAM,KAAK,GAAG,EAAE,GAAG,GAAG,GAAG,GAAG,CAAC;YAC7B,MAAM,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC;YAC/B,MAAM,MAAM,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC;YAChC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;YAC3B,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;YACzB,MAAM,IAAI,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC;YACvB,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;YAC7B,IAAI,CAAC;gBACH,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;gBAChD,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC/C,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;gBACxC,EAAE,CAAC,YAAY,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;gBAC5F,OAAO,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;YAChD,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;gBACnB,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBACjB,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACf,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;QAED,aAAa,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI;YAClG,MAAM,KAAK,GAAG,EAAE,GAAG,GAAG,GAAG,GAAG,CAAC;YAC7B,MAAM,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC;YAC/B,MAAM,MAAM,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC;YAChC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;YAC3B,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;YACzB,MAAM,IAAI,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC;YACvB,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;YAC7B,IAAI,CAAC;gBACH,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;gBAChD,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC/C,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;gBACxC,EAAE,CAAC,cAAc,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;gBACnG,OAAO,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;YAChD,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;gBACnB,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBACjB,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACf,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;QAED,gBAAgB,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI;YAC5E,MAAM,KAAK,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,CAAC;YAC9B,MAAM,IAAI,GAAG,IAAI,GAAG,GAAG,GAAG,EAAE,GAAG,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;YAClC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;YAC3B,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;YACzB,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;YACzB,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;YAC7B,IAAI,CAAC;gBACH,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;gBAChD,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC/C,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC5C,EAAE,CAAC,kBAAkB,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC9E,OAAO,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;YAChD,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;gBACnB,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBACjB,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBACjB,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;QAED,iBAAiB,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,gBAAgB;YAC7G,MAAM,KAAK,GAAG,GAAG,GAAG,GAAG,GAAG,GAAG,CAAC;YAC9B,MAAM,IAAI,GAAG,IAAI,GAAG,GAAG,GAAG,EAAE,GAAG,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;YACnC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;YAC3B,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;YACzB,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC;YAC/B,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;YAC7B,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;YAC7B,MAAM,MAAM,GAAG,gBAAgB,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACnD,IAAI,CAAC;gBACH,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;gBAChD,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC/C,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;gBACrD,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;gBACjD,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;gBACjD,EAAE,CAAC,mBAAmB,CACpB,KAAK,EACL,GAAG,EACH,GAAG,EACH,IAAI,EACJ,OAAO,EACP,IAAI,EACJ,IAAI,EACJ,QAAQ,EACR,QAAQ,EACR,MAAM,EACN,GAAG,EACH,IAAI,EACJ,EAAE,EACF,EAAE,EACF,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CACzB,CAAC;gBACF,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC/B,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC/B,OAAO,gBAAgB,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YACzE,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;gBACnB,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBACjB,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBACvB,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;gBACrB,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;gBACrB,IAAI,MAAM,EAAE,CAAC;oBACX,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;gBACtB,CAAC;YACH,CAAC;QACH,CAAC;QAED,aAAa,CAAC,KAAK,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI;YAC/F,MAAM,KAAK,GAAG,QAAQ,GAAG,GAAG,GAAG,GAAG,CAAC;YACnC,MAAM,IAAI,GAAG,IAAI,GAAG,KAAK,CAAC;YAC1B,MAAM,IAAI,GAAG,IAAI,GAAG,KAAK,CAAC;YAC1B,MAAM,MAAM,GAAG,QAAQ,GAAG,IAAI,GAAG,IAAI,CAAC;YACtC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;YAC3B,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;YAC5B,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;YAC1B,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;YAC5B,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;YAC1B,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;YAC7B,IAAI,CAAC;gBACH,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;gBAChD,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC/C,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;gBACjD,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC/C,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;gBACjD,EAAE,CAAC,cAAc,CAAC,KAAK,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;gBAC/G,OAAO,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;YAChD,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;gBACnB,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;gBACpB,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;gBAClB,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;gBACpB,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;gBAClB,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;QAED,UAAU,CAAC,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM;YAC1C,MAAM,IAAI,GAAG,MAAM,GAAG,KAAK,CAAC;YAC5B,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,CAAC;YACxB,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;YAC7B,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;YAC7B,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;YAC7B,IAAI,CAAC;gBACH,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC;gBACjD,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;gBACjD,EAAE,CAAC,WAAW,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;gBAChE,OAAO,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;YAChD,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBACrB,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;gBACrB,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,IAAI,QAAQ,GAAG,KAAK,CAAC;AACrB,IAAI,OAAgC,CAAC;AAErC,MAAM,UAAU,OAAO;IACrB,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,QAAQ,GAAG,IAAI,CAAC;QAChB,MAAM,EAAE,GAAG,WAAW,EAAE,CAAC;QACzB,OAAO,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACvC,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,WAAW;IACzB,QAAQ,GAAG,IAAI,CAAC;IAChB,OAAO,GAAG,SAAS,CAAC;AACtB,CAAC;AAED,MAAM,UAAU,eAAe;IAC7B,OAAO,OAAO,EAAE,KAAK,SAAS,CAAC;AACjC,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@kongyo2/kongyo2x",
|
|
3
|
-
"version": "0.
|
|
4
|
-
"description": "Image super-resolution in TypeScript: MLPconv networks trained from scratch
|
|
3
|
+
"version": "0.3.1",
|
|
4
|
+
"description": "Image super-resolution in TypeScript with a Rust/WebAssembly compute core: MLPconv networks trained from scratch, with a pure-TypeScript fallback.",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"license": "MIT",
|
|
7
7
|
"author": "kongyo2",
|
|
@@ -22,7 +22,10 @@
|
|
|
22
22
|
"brain.js",
|
|
23
23
|
"mlpconv",
|
|
24
24
|
"neural-network",
|
|
25
|
-
"typescript"
|
|
25
|
+
"typescript",
|
|
26
|
+
"webassembly",
|
|
27
|
+
"wasm",
|
|
28
|
+
"rust"
|
|
26
29
|
],
|
|
27
30
|
"bin": {
|
|
28
31
|
"kongyo2x": "dist/cli.js"
|
|
@@ -38,6 +41,7 @@
|
|
|
38
41
|
"files": [
|
|
39
42
|
"dist",
|
|
40
43
|
"models",
|
|
44
|
+
"wasm/kongyo2x.wasm",
|
|
41
45
|
"LICENSE"
|
|
42
46
|
],
|
|
43
47
|
"engines": {
|
|
@@ -48,6 +52,7 @@
|
|
|
48
52
|
},
|
|
49
53
|
"scripts": {
|
|
50
54
|
"build": "tsc",
|
|
55
|
+
"build:wasm": "bash scripts/build-wasm.sh",
|
|
51
56
|
"prepublishOnly": "npm run build",
|
|
52
57
|
"typecheck": "tsc --noEmit",
|
|
53
58
|
"typecheck:watch": "tsc --noEmit --watch",
|
|
Binary file
|