pytorch model
的前端运行。
参考资料
使用
首先把项目 clone
下来,然后在建立文件夹 src/assets/
文件夹。
接下来,我们就要生成 model.onnx
,可以直接里面的 onnx_model.py
文件,会自动下载 pth
模型。根据文档中所说,该模型是 alex
模型。
由于,py
下载可能很慢,所以,可以直接从链接上下载。
下载下来的文件中,放在缓存位置。「该位置在你运行 py
文件的时候会给出」
至于什么是 alex
模型,请参考
下载下来模型之后,运行 py
文件,会自动转成 onnx
模型,然后放在 src/assets/
中。
然后直接执行
npm run serve
就能够在页面上看到效果了。