yizhi-js-lib/ai-box
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

代码优化

Browse Source
This commit is contained in:
Yizhi
2025-03-10 18:12:32 +08:00
parent a966b82963
commit 524dcaecbd
10 changed files with 71 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/backend/mnn/session.ts
View File

@ -1,3 +1,4 @@
import { getConfig } from "../../config";
import { CommonSession, dataTypeFrom, isTypedArray, SessionNodeData, SessionNodeInfo, SessionRunInputOption, SessionRunOutput } from "../common";
export class MNNSession extends CommonSession {
@ -7,7 +8,7 @@ export class MNNSession extends CommonSession {
public constructor(modelData: Uint8Array) {
super();
const addon = require("../../../build/mnn.node")
const addon = require(getConfig("MNN_ADDON_FILE"));
this.#session = new addon.MNNSession(modelData);
}

3
src/backend/ort/session.ts
View File

@ -1,3 +1,4 @@
import { getConfig } from "../../config";
import { CommonSession, dataTypeFrom, isTypedArray, SessionNodeData, SessionNodeInfo, SessionRunInputOption, SessionRunOutput } from "../common";
export class OrtSession extends CommonSession {
@ -7,7 +8,7 @@ export class OrtSession extends CommonSession {
public constructor(modelData: Uint8Array) {
super();
const addon = require("../../../build/ort.node")
const addon = require(getConfig("ORT_ADDON_FILE"));
this.#session = new addon.OrtSession(modelData);
}

13
src/config.ts Normal file
View File

@ -0,0 +1,13 @@
import path from "path";
const defaultAddonDir = path.join(__dirname, "../build")
const aiConfig = {
"CV_ADDON_FILE": path.join(defaultAddonDir, "cv.node"),
"MNN_ADDON_FILE": path.join(defaultAddonDir, "mnn.node"),
"ORT_ADDON_FILE": path.join(defaultAddonDir, "ort.node"),
};
export function setConfig<K extends keyof typeof aiConfig>(key: K, value: typeof aiConfig[K]) { aiConfig[key] = value; }
export function getConfig<K extends keyof typeof aiConfig>(key: K): typeof aiConfig[K] { return aiConfig[key]; }

3
src/cv/mat.ts
View File

@ -1,3 +1,4 @@
import { getConfig } from "../config";
export enum ImreadModes {
IMREAD_UNCHANGED = -1,
@ -32,7 +33,7 @@ export class Mat {
}
public constructor(imageData: Uint8Array, option?: MatConstructorOption) {
const addon = require("../../build/cv.node");
const addon = require(getConfig("CV_ADDON_FILE"));
if ((imageData as any) instanceof addon.Mat) this.#mat = imageData;
else this.#mat = new addon.Mat(imageData, option);
}

4
src/index.ts
View File

@ -0,0 +1,4 @@
import * as ai from "./main";
export default ai;
export { ai };

4
src/main.ts
View File

@ -0,0 +1,4 @@
export { deploy } from "./deploy";
export { cv } from "./cv";
export { backend } from "./backend";
export { setConfig as config } from "./config";

1
src/test.ts
View File

@ -4,6 +4,7 @@ import { cv } from "./cv";
import { faceidTestData } from "./test_data/faceid";
import path from "path";
import crypto from "crypto";
import ai from ".";
async function cacheImage(group: string, url: string) {
const _url = new URL(url);

33
src/utils/utils.ts
View File

@ -1,33 +0,0 @@
export namespace utils {
export function rgba2rgb<T extends Uint8Array | Float32Array>(data: T): T {
const pixelCount = data.length / 4;
const result = new (data.constructor as any)(pixelCount * 3) as T;
for (let i = 0; i < pixelCount; i++) {
result[i * 3 + 0] = data[i * 4 + 0];
result[i * 3 + 1] = data[i * 4 + 1];
result[i * 3 + 2] = data[i * 4 + 2];
}
return result;
}
export function rgb2bgr<T extends Uint8Array | Float32Array>(data: T): T {
const pixelCount = data.length / 3;
const result = new (data.constructor as any)(pixelCount * 3) as T;
for (let i = 0; i < pixelCount; i++) {
result[i * 3 + 0] = data[i * 3 + 2];
result[i * 3 + 1] = data[i * 3 + 1];
result[i * 3 + 2] = data[i * 3 + 0];
}
return result;
}
export function normalize(data: Uint8Array | Float32Array, mean: number[], std: number[]): Float32Array {
const result = new Float32Array(data.length);
for (let i = 0; i < data.length; i++) {
result[i] = (data[i] - mean[i % mean.length]) / std[i % std.length];
}
return result;
}
}