0%

开源项目 | js onnxruntime-web-tutorial

pytorch model 的前端运行。

参考资料

使用

首先把项目 clone 下来,然后在建立文件夹 src/assets/ 文件夹。

接下来,我们就要生成 model.onnx ,可以直接里面的 onnx_model.py 文件,会自动下载 pth 模型。根据文档中所说,该模型是 alex 模型。

由于,py 下载可能很慢,所以,可以直接从链接上下载。

下载下来的文件中,放在缓存位置。「该位置在你运行 py 文件的时候会给出」

至于什么是 alex 模型,请参考

下载下来模型之后,运行 py 文件,会自动转成 onnx 模型,然后放在 src/assets/ 中。

然后直接执行

npm run serve

就能够在页面上看到效果了。

请我喝杯咖啡吧~