可以使用ONNX
先使用ONNX将pytorch或则TensorFlow训练好的模型转成onnx格式文件
pip install onnx
imprt torch import torchvision # Load the trained PyTorch model model = torchvision.models.resnet18(pretrained=True) # Export the model to ONNX format input_shape = (1, 3, 224, 224) torch.onnx.export(model, torch.randn(input_shape), "model.onnx", verbose=True)
然后再利用onnx.js在浏览器中载入onnx格式模型文件,就能在浏览器运行ai模型了。
npm install onnxjs
// 1. Load the model const model = await onnx.load("model.onnx"); // 2. Prepare input data const inputData = new Float32Array([1, 2, 3, 4]); // 3. Run the model const output = await model.execute({ input: inputData }); // 4. Process the output console.log(output);
当然tensorflow.js也可以实现在浏览器中运行TensorFlow训练的模型文件。
网友回复
openai发布的agentkit与coze扣子、dify等流程搭建智能体有啥不同?
阿里云上的ecs镜像存储还要钱,如何免费下载到本地以后再创建?
如何通过调用大模型api实现输入一个商品图片生成模特展示解说的宣传短片?
qwen千问大模型api如何内置互联网搜索?
YOLO如何结合opencv实现视觉实时摔倒检测?
html中内嵌style与link引入css代码报错的处理机制不同?
coze扣子中调用seadream4多参考图修改ps图片如何返回图片尺寸设为第一个图片的尺寸?
coze扣子中如何将原图缩放到指定尺寸?
如何解决传大文件突然断网重传的问题?
CefSharp与Electron开发桌面应用哪个更好?