#AI工具箱 ## 安装 ``` npm install @yizhi/ai ``` ## 使用 ```typescript import ai from "@yizhi/ai"; // 配置相应的node插件,插件需自行编译 ai.config("CV_ADDON_FILE", "/path/to/cv.node"); ai.config("MNN_ADDON_FILE", "/path/to/mnn.node"); ai.config("ORT_ADDON_FILE", "/path/to/onnxruntime.node"); //直接推理 const facedet = await ai.deploy.facedet.Yolov5Face.load("YOLOV5S_ONNX"); const boxes = await facedet.predict("/path/to/image"); //使用自己的模型 const session = new ai.backend.ort.Session(modelBuffer); const outputs = session.run(inputs); ``` ## 插件编译 1. 依赖 1. cmake 1. ninja 1. c++编译器(gcc,clang,Visual Studio ...) 1. 编译第三方库 ``` node thirdpart/install.js --with-mnn --with-onnx ``` 1. 编译插件 ``` cmake -B build -G Ninja . -DCMAKE_BUILD_TYPE=Release cmake --build build --config Release ``` 注意:注意:在Windows下编译时,需要打开Visual Studio命令行